在進行語義分割的時候,使用png圖像進行圖像的語義掩碼圖像進行語義分割信息的存儲,但是實際存儲的LabelID是比較小的數字,直接進行存儲的時時候,打開的語義掩碼圖像是黑色的,不方便進行查看顯示等。 所以需要一種存儲方式,即方便展示又方便在訓練的時候進行讀取。即Pascal的語義信息的存儲方式,帶有調色板的彩色圖,P模式的彩色圖,這裏記錄如何使用python進行P模式圖像的製作。
from PIL import Image
import PIL.ImageDraw as ImageDraw
im = Image.new("P", (400, 400), 0)
im.putpalette([
0, 0, 0, # black background
255, 0, 0, # index 1 is red
255, 255, 0, # index 2 is yellow
255, 153, 0, # index 3 is orange
])
d = ImageDraw.ImageDraw(im)
d.polygon((0, 0, 0, 400, 400, 400),fill=1)
d.rectangle((100, 100, 300, 300),fill=3)
d.ellipse((120, 120, 280, 280),fill=2)
im.save("out.gif")
繪製的結果如下:
參考資料:
[1] Creating Palette Images,http://effbot.org/zone/creating-palette-images.htm