CudaImage.h struct CudaImage { int width; int height; float *imData; float *cuData; size_t pitch; size_t nBytes; void setPixel(int x, int y, float val); //设置(x,y)处的像素值为val float getPixel(int x, int y); //取得(x,y)处坐标的像素值 float getPixelBI(float col, float row); //也是取得(x,y)处坐标的像素值,不过此时x,y都是浮点数,这个像素值是由它周围4个整数点的像素值通过距 离上的权值得到的。 void InitCuData(int _width, int _height); //初始化显存 /*InitCuData函数用到了一个cudaMallocPitch(),说明向设备分配至少widthInBytes*height字节的线性存储器 ,并以*devPtr的形式返回指向所分配存储器的指针。该函数可以填充所分配的存储器,以确保在地址从一行更新到另一行时,给定行的对应指针依然满足对齐要求。*/
cudaMallocPitch((void**)(&cuData), &pitch, sizeof(float) * width, height);
/* *devPtr的形式返回指向所分配存储器的指针。以*pitch的形式返回间距,即所分配存储器的宽度,以字节为 单位。分配至少widthInBytes*height字节的线性存储器。*/ void InitImData(int _width, int _height); //imData是一个 大小width*height的浮点数组,用来在主机上存储图像数据的结构 void CopyToDevice(); //把数据从cpu上的imData复制到gpu上,返回指针cuData。使用了cudaMemcpy2D这个api,用于分配内存。
cudaMemcpy2D(cuData, pitch, imData, sizeof(float) * width, sizeof(float) * width, height,
cudaMemcpyHostToDevice)); void CopyToHost(); //把数据从gpu上cuData指针复制到主机上imData int LoadImage(const char *filename); /*加载图片文件,数据存入imData。首先把图片用opencv的函数cvLoadImage加载到opencv的结构,再从opencv 的结构中把数据转成我们的CudaImage。*/ void toIplImage(IplImage *&img); //把imData数据转成opencv格式 IplImage void write(const char * filename); //把imData的数据写入文件中,比如写成.jpg的文件 void show(const char * windowname); //把imData显示在屏幕上 }
CUDA 图像数据结构
最新推荐文章于 2023-05-29 17:53:39 发布