python - 将 RGB 中的随机颜色分配给图像

Question

我正在用 Python 使用 OpenCV 实现一张图片，要求是为图像中的每个像素分配随机的 R、G、B 颜色。

这是我当前的代码:

red, green, blue = (255, 0, 0), (0, 255, 0), (0, 0, 255)
rgb = [red, green, blue]
image = img_gen(width, height, rgb_color = random.choice(rgb))

width, height = 640, 480

def img_gen(width, height, rgb_color=(0, 0, 0)):
  ...
    for x in range (0, width):
      for y in range (0, height):
          # what code should be inserted here?

我现在可以实现一张图片，只分配一个随机选择来填充整个图像，但我想知道如何访问每个“像素”。

Answer 1

使用 numpy，可以轻松创建具有随机颜色像素的图像。为了与 OpenCV 一起使用，您通常希望数据类型为 uint8(从 0 到 255 的整数):

import numpy as np
width, height = 640, 480
img = np.round(np.random.rand(height, width, 3) * 255).astype(np.uint8)
print(img.shape)  # (480, 640, 3)
print(img.dtype)  # uint8

注意 numpy 在宽度之前使用高度 :)