make_image_data.py

# generate n images of random, flat color and save to disk:
n = 100
output_dir = 'rgb_images'
import os
import random
from PIL import Image
os.makedirs(output_dir, exist_ok=True)
for i in range(n):
    img = Image.new('RGB', (1024, 1024), (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
    img.save(output_dir + '/image' + str(i) + '.jpg')