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

Question

I am implement a picture using OpenCV with Python, the requirement is to assign random R,G,B colour to every pixel into the image.

Here is my current code:

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?

I can now implement a picture with assigning only one random choice to fill in the whole image, but I want to know hot to access to every "pixel".

Answer 1

Using numpy, it's easy to create an image with randomly colored pixels. For use with OpenCV, you often want the data type to be uint8 (integers from 0 to 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

Note that numpy uses height before width :)

Answer 2

I'm not exactly sure what your img_gen function returns, but to get a random entry from rgb you can use

be sure to import random try :

img = []
for x in range (0, width):
    for y in range (0, height):
        img.append(rgb[random.randint(0,2)])
return img

It really depends on the image format openCV expects in the functions you want to use.

edit :

if you are using numpy you should use

img = np.array(width*height)

then refer to the content with

img[x*width + y] = ....

Answer 3

You can simply use -

img = np.random.rand(width, height, 3);

This will create random pixel values.

CAVEAT - RGB values will be between [0,1] and not [0,255].