我正在使用 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 。我该怎么做?
最佳答案
虽然这最终取决于图像的存储方式,但在这种情况下我会利用结构。例如:
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
关于c - 如何读取 C 中图像的 RGB channel ?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/13421220/