c - 如何读取 C 中图像的 RGB channel ？

Question

我正在使用 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 channel 。我该怎么做？

Answer 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