CUDAとOpenCVを使用して1つのイメージチャネルを削除する

Question

OpenCVの学習を始めたばかりです。 CUDAカーネルを使用して1つのチャンネルを削除し、それが元の画像にどのような影響を与えるかを視覚化したいと思います。 しかし、プログラムは動作しません、理由は分かりません。それはちょうど黒いウィンドウを示し:(ここ はコードです：

#include "opencv2\opencv.hpp" 
#include <cuda.h> 
#include "cuda_runtime.h" 
#include "device_launch_parameters.h" 
#include <iostream> 
#include <device_functions.h> 


using namespace cv; 


__global__ void imgProc(unsigned char *in, unsigned char * out) 
{ 
    int i = threadIdx.x + blockIdx.x * blockDim.x; 
    out[i] =in[i]; 
    out[i+1] = in[i+1]; 
    out[i + 2] = 0; //deleting one channel 


} 

int main() 
{ 
    Mat file1 = imread("sw.jpg", CV_LOAD_IMAGE_COLOR); 
    unsigned char *input = (unsigned char*)(file1.data); 
    unsigned char *dev_input, *dev_output; 
    unsigned char *output = (unsigned char*)malloc(file1.cols*file1.rows * 3 * sizeof(char)); 

    cudaMalloc((void**)&dev_input, file1.cols*file1.rows * 3 * sizeof(char)); 
    cudaMalloc((void**)&dev_output, file1.cols*file1.rows * 3 * sizeof(char)); 
    cudaMemcpy(dev_input, input, file1.cols*file1.rows * 3 * sizeof(char), cudaMemcpyHostToDevice); 
    imgProc << <file1.cols, file1.rows >> > (dev_input, dev_output); 
    cudaMemcpy(output, dev_output, file1.cols*file1.rows * 3 * sizeof(char), cudaMemcpyDeviceToHost); 

    Mat file3 = Mat(file1.rows,file1.cols, CV_8UC3,output); 
    namedWindow("Modified", CV_WINDOW_FREERATIO); 
    imshow("Modified", file3); 
    namedWindow("Original", CV_WINDOW_FREERATIO); 
    imshow("Original", file1); 

    cudaFree(dev_input); 
    cudaFree(dev_output); 
    free(output); 


    waitKey(); 

    return 0; 
} 

Answer 1

あなたがOpenCVのは、このタスクを完了するために必要なすべての機能を提供し、それが必要以上に、これはより複雑になっているように見える：

split(src,BGRChannels); // split the BGR channesl 
BGRChannels[1]=Mat::zeros(src.rows,src.cols,CV_8UC1);// removing Green channel 
merge(BGRChannels,3,src); // pack the image 

Answer 2

[OK]を、私は、カーネル内のいくつかのミスがありました。それを得たが、私は働いていた絵の中で最も重要なのはサイズが私のGPU上のグリッドごとの最大スレッド数よりも大きかった。

Here is working code, which deletes one img channel from the picture: 

#include "opencv2\opencv.hpp" 
#include <cuda.h> 
#include "cuda_runtime.h" 
#include "device_launch_parameters.h" 
#include <iostream> 
#include <device_functions.h> 


using namespace cv; 


__global__ void imgProc(unsigned char *in, unsigned char * out) 
{ 
    int x = blockIdx.x; 
    int y = blockIdx.y; 
    int offset = x + y * gridDim.x; 
    out[offset*3+0] =0; 
    out[offset * 3 + 1] = in[offset * 3 + 1]; 
    out[offset * 3 + 2] = in[offset * 3 + 2]; 


} 

int main() 
{ 
    cudaDeviceProp prop; 
    cudaGetDeviceProperties(&prop, 0); 
    std::cout << (int)prop.maxGridSize[1]; 


    Mat file1 = imread("sw.jpg", CV_LOAD_IMAGE_COLOR); 
    unsigned char *input = (unsigned char*)(file1.data); 
    unsigned char *dev_input, *dev_output; 
    unsigned char *output = (unsigned char*)malloc(file1.cols*file1.rows * 3 * sizeof(char)); 

    cudaMalloc((void**)&dev_input, file1.cols*file1.rows * 3 * sizeof(char)); 
    cudaMalloc((void**)&dev_output, file1.cols*file1.rows * 3 * sizeof(char)); 
    cudaMemcpy(dev_input, input, file1.cols*file1.rows * 3 * sizeof(char), cudaMemcpyHostToDevice); 

    dim3 grid(file1.cols, file1.rows); 
    imgProc << <grid,1 >> > (dev_input, dev_output); 
    cudaMemcpy(output, dev_output, file1.cols*file1.rows * 3 * sizeof(char), cudaMemcpyDeviceToHost); 

    Mat file3 = Mat(file1.rows,file1.cols, CV_8UC3,output); 
    namedWindow("Modified", CV_WINDOW_FREERATIO); 
    imshow("Modified", file3); 
    namedWindow("Original", CV_WINDOW_FREERATIO); 
    imshow("Original", file1); 

    cudaFree(dev_input); 
    cudaFree(dev_output); 
    free(output); 


    waitKey(); 

    return 0; 
}