【PIL】——PIL灰度图以调色板保存成彩图

目的： 为了可视化语义分割的label图像

Code

import numpy as np
from PIL import Image

label = Image.open('label.png')
bin_colormap = np.random.randint(0, 255, (256, 3))      # 可视化的颜色
bin_colormap = bin_colormap.astype(np.uint8)
visualimg  = Image.fromarray(label, "P")
palette = bin_colormap          #long palette of 768 items
visualimg.putpalette(palette) 
visualimg.save(visualname, format='PNG')

可视化效果