實施OpenCV函數

Question

我有格式32FC1的像素陣列（32位浮點與一個信道點）

float * image; // my original image on the GPU 
float * image_d; 
int size = row*col*sizeof(float); 
cudaMalloc((void**)&f_d, size); 
cudaMemcpy(image_d, image, size, cudaMemcpyHostToDevice); 

現在，我想呼叫的功能gpu::GaussianBlur來自OpenCV image_d，但此功能只接受格式爲"GpuMat"的參數。有什麼辦法可以將GPU上的array轉換爲GpuMat？

我的意思是，我不想在CPU和GPU之間發生任何內存複製。我希望OpenCV函數可以直接處理GPU上的陣列。

Answer 1

您可以創建GpuMat形成設備指針：

const size_t step = col * sizeof(float); 
GpuMat mat(row, col, CV_32FC1, image_d, step); 

這GpuMat不會釋放內存，你應該手動調用cudaFree。