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()

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章