这里用到的是cv2和numpy的组合。cv2会把图片都处理成uint8的numpy矩阵,同理,我们可以把符合图片通道要求的numpy uint8矩阵用cv2保存成图片。
直接看代码:
import numpy as np
import cv2
color = (255, 128, 76) # color you want
arr = np.zeros((224, 224, 3), dtype=np.uint8) # all-zero array
pic = color - arr
cv2.imwrite('demo.jpg', pic)
直接在第三行定义出你想要的图片颜色即可。
可以加入随机数:
import numpy as np
import cv2
from random import random
color = (int(255 * random()), int(255 * random()), int(255 * random())) # random color
arr = np.zeros((224, 224, 3), dtype=np.uint8) # all-zero array
pic = color - arr
cv2.imwrite('demo.jpg', pic)