語義分割掩碼圖像-Palette Images

在進行語義分割的時候,使用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

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