CUDA 图像数据结构

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显示在屏幕上 }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值