Blame view

tsl_aiplatform/reprocessing_module/CropImg.h 599 Bytes
85cc8cb9   Hu Chunming   ๅŽŸ็‰ˆไปฃ็ 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
  #ifndef _CROP_GPUIMG_
  #define _CROP_GPUIMG_
  
  #include <cuda.h>
  #include <cuda_runtime.h>
  #include <device_launch_parameters.h>
  
  namespace cudacommon {
  
  	//void CropImgGpu(const float * d_srcRGB, const int src_width, const int src_height, float* d_dstRGB, int left, int top, int width, int height);
  
  	cudaError_t CropImgGpu(const unsigned char * d_srcRGB, const int src_width, const int src_height, unsigned char* d_dstRGB, int left, int top, int width, int height);
  
  	void ResizeImgGpu(const float* d_srcRGB, int src_width, int src_height, float* d_dstRGB, int dst_width, int dst_height);
  }
  
  
  #endif