【脚本】灰度图转P模式图的核心代码

实现效果

import imgviz
from PIL import Image

if __name__ == '__main__':
    # 1. 先用PIL读入
    imagepath = r'E:\A_new_segdet\deeplabv3-plus-pytorch-main\miou_out\detection-results\crack_8.png'
    image = Image.open(imagepath)
    # 2. 转为P模式
    image = image.convert('P')
    # 3. 调色板上色
    colormap = imgviz.label_colormap()
    image.putpalette(colormap.flatten())

    image.save('image.png')