c - 如何在 C 中读取图像的 RGB 通道？

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 通道。我该怎么做？

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