在 thrust 中调用用户定义的函数

Question

我正在使用 OpenCV 加载一个 .png 文件，我想使用 thrust 库提取它的蓝色强度值。

我的代码是这样的:

1. 使用 OpenCV IplImage 指针加载图像
2. 将图像数据复制到thrust::device_vector
3. 使用推力库从结构内的设备 vector 中提取蓝色强度值。

现在我在从设备 vector 中提取蓝色强度值时遇到了问题。

• 我在 cuda 中编写了这段代码，现在已经使用推力库对其进行了转换。
• 我在此函数中获取蓝色强度值。
• 我想知道如何从主函数调用此结构 FetchBlueValues。

代码:

#define ImageWidth 14
#define ImageHeight 10

thrust::device_vector<int> BinaryImage(ImageWidth*ImageHeight);
thrust::device_vector<int> ImageVector(ImageWidth*ImageHeight*3);

struct FetchBlueValues
{
    __host__ __device__ void operator() ()
    {
        int index = 0 ;
        for(int i=0; i<= ImageHeight*ImageWidth*3 ; i = i+3)
        {
            BinaryImage[index]= ImageVector[i];
            index++;
        }
    }
};

void main()
{
    src = cvLoadImage("../Input/test.png", CV_LOAD_IMAGE_COLOR);

    unsigned char *raw_ptr,*out_ptr;
    raw_ptr = (unsigned char*) src->imageData;

    thrust::device_ptr<unsigned char> dev_ptr = thrust::device_malloc<unsigned char>(ImageHeight*src->widthStep);

    thrust::copy(raw_ptr,raw_ptr+(src->widthStep*ImageHeight),dev_ptr);
    int index=0;
    for(int j=0;j<ImageHeight;j++)
    {
        for(int i=0;i<ImageWidth;i++)
        {
            ImageVector[index] = (int) dev_ptr[ (j*src->widthStep) + (i*src->nChannels) + 0 ];
            ImageVector[index+1] = (int) dev_ptr[ (j*src->widthStep) + (i*src->nChannels) + 1 ];
            ImageVector[index+2] = (int) dev_ptr[ (j*src->widthStep) + (i*src->nChannels) + 2 ];

            index +=3 ;
        }
    }

}

Answer 1

由于图像以像素格式存储，并且每个像素都包含不同的颜色，因此在访问每个像素的各个颜色分量时有一个自然的“步幅”。在这种情况下，一个像素的颜色分量似乎存储在每个像素三个连续的 int 数量中，因此给定颜色分量的访问步幅为三个。

涵盖了跨范围访问迭代器方法的示例 here .