CUDA二维纹理内存+OpenCV图像滤波

来源：互联网发布：多功能拍照软件编辑：程序博客网时间：2024/05/14 11:04

CUDA和OpenCV混合编程，使用CUDA的纹理内存，实现图像的二值化以及滤波功能。

#include <cuda_runtime.h> #include <highgui/highgui.hpp>#include <imgproc/imgproc.hpp>using namespace cv;int width = 512;int height = 512;// 2维纹理texture<float, 2, cudaReadModeElementType> texRef;// 核函数__global__ void transformKernel(uchar* output, int width, int height){// 根据tid bid计算归一化的拾取坐标unsigned int x = blockIdx.x * blockDim.x + threadIdx.x;unsigned int y = blockIdx.y * blockDim.y + threadIdx.y;float u = x / (float)width;float v = y / (float)height;//从纹理存储器中拾取数据，并写入显存output[(y * width + x)] = tex2D(texRef, u / 4, v / 4);}int main(){// 分配CUDA数组cudaChannelFormatDesc channelDesc = cudaCreateChannelDesc(32, 0, 0, 0, cudaChannelFormatKindFloat);cudaArray* cuArray;cudaMallocArray(&cuArray, &channelDesc, width, height);//使用OpenCV读入图像Mat image = imread("D:\\lena.jpg", 0);resize(image, image, Size(width, height));imshow("原始图像", image);cudaMemcpyToArray(cuArray, 0, 0, image.data, width*height, cudaMemcpyHostToDevice);// 设置纹理属性texRef.addressMode[0] = cudaAddressModeWrap; //循环寻址方式texRef.addressMode[1] = cudaAddressModeWrap;texRef.filterMode = cudaFilterModeLinear;   //线性滤波texRef.normalized = true; //归一化坐标 //绑定纹理cudaBindTextureToArray(texRef, cuArray, channelDesc);Mat imageOutput = Mat(Size(width, height), CV_8UC1);uchar * output = imageOutput.data;cudaMalloc((void**)&output, width * height * sizeof(float));dim3 dimBlock(16, 16);dim3 dimGrid((width + dimBlock.x - 1) / dimBlock.x, (height + dimBlock.y - 1) / dimBlock.y);transformKernel << <dimGrid, dimBlock >> > (output, width, height);cudaMemcpy(imageOutput.data, output, height*width, cudaMemcpyDeviceToHost);imshow("CUDA+OpenCV滤波", imageOutput);waitKey();cudaFreeArray(cuArray);cudaFree(output);}

原始lena图像：

运行效果：

0 0