Create pure color image by given RGB values.

colorimg.py

"""
Create pure color JPEG by given RGB values.
"""

import cv2
import numpy as np


def paint_it_color(rgb=(0, 0, 0), shape=(720, 1280, 3), filename='black'):
    img = np.zeros(shape)
    img[:, :, 0] = np.ones((shape[0], shape[1])) * rgb[2]
    img[:, :, 1] = np.ones((shape[0], shape[1])) * rgb[1]
    img[:, :, 2] = np.ones((shape[0], shape[1])) * rgb[0]
    cv2.imwrite(filename + '.jpg', img)


def paint_it_black(shape=(720, 1280, 3)):
    paint_it_color(rgb=(0, 0, 0), shape=shape, filename='black')


if __name__ == '__main__':
    paint_it_black(shape=(1080, 1920, 3))