python画一张纯色图

做语义分割的时候，得到的一般都是8位灰色图，利用Numpy可以完成非前景部分全部置为纯色图，或者透明化。

1、PIL库完成

除了这里介绍到的用opencv和numpy，还有人用PIL库完成的：用python画一张红色的1080p图片。如果想转换成opencv可用的格式，参考python中PIL.Image和OpenCV图像格式相互转换即可。

from PIL import Image
import cv2 as cv
import numpy as np
img = Image.new("RGB",(1920,1080),(255,0,255))
img = cv.cvtColor(np.asarray(img), cv.COLOR_RGB2BGR)
cv.imshow("white", img)
cv.waitKey()

2、opencv + numpy

因为平时用opencv较多，现在介绍一个opencv的方法，看代码如下：

import cv2 as cv
import numpy as np
# 图像由三通道组成，bgr三通道组合形成即可。
b_channel = np.ones((1080, 1920), dtype=np.uint8) * 0
g_channel = np.ones((1080, 1920), dtype=np.uint8) * 100
r_channel = np.ones((1080, 1920), dtype=np.uint8) * 200
image = cv.merge((b_channel, g_channel, r_channel))
cv.imshow("image", image)
cv.waitKey()