1

我正在.png使用 OpenCV 加载文件，我想使用推力库提取其蓝色强度值。

我的代码是这样的：

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

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

  • 我已经在 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 ;
        }
    }

}
4

1 回答 1

1

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

此处介绍了一个示例跨步范围访问迭代器方法。

于 2013-07-18T00:11:01.567 回答