1

我正在使用 pycuda 来计算图像像素的强度。为此,我将图像发送到 GPU,如下所示。

 img = np.float32(np.array(img.imread('my_image.jpg')))
 img_gpu = gpuarray.to_gpu(img)

然后,在内核中(当然是用 c 编写的),我想得到如下的 RGB 值(在伪代码中)。

__global__ void get_intensities(float* img, float* intensities) {
    intensities[globalIndex] = R(x, y) + G(x, y) + B(x, y)
}

我现在的大问题是在 C 中获取 RGB 通道。我该怎么做?

4

1 回答 1

1

尽管这最终取决于图像的存储方式,但在这种情况下我会利用结构。例如:

image = np.uint8(np.array(img.imread('my_image.jpg')))
img_gpu = gpuarray.to_gpu(image)

__global__ void intensity(uchar3* img, int* intensities)
{
    ...
    intensities[globalIndex] = img[globalIndex].x + img[globalIndex].y + img[globalIndex].z;
    ...
}

请参阅http://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#built-in-vector-types

于 2012-11-21T20:33:23.807 回答