VOC2012 数据集保存成彩色调色盘 P模式 PIL Image

末蓝、 2021-12-16 04:26 226阅读 0赞

VOC的标注22个，从0-21，其中21 是未标注类，不过实际上读取出来是255，所以你的标签\[0,...20,255\]

out数据的要求是int,数值在\[0,...20,255\]

代码：

import numpy as np
    from PIL import Image
    
    
    # # load image, switch to BGR, subtract mean, and make dims C x H x W for Caffe
    # im = Image.open('data/pascal/VOCdevkit/VOC2012/JPEGImages/2007_000129.jpg')
    # in_ = np.array(im, dtype=np.float32)
    # in_ = in_[:,:,::-1]
    # in_ -= np.array((104.00698793,116.66876762,122.67891434))
    # in_ = in_.transpose((2,0,1))
    #
    # # load net
    # net = caffe.Net('voc-fcn8s/deploy.prototxt', 'voc-fcn8s/fcn8s-heavy-pascal.caffemodel', caffe.TEST)
    # # shape for input (data blob is N x C x H x W), set data
    # net.blobs['data'].reshape(1, *in_.shape)
    # net.blobs['data'].data[...] = in_
    # # run net and take argmax for prediction
    # net.forward()
    # out = net.blobs['score'].data[0].argmax(axis=0)
    def colorful(out,name):
        arr=out.astype(np.uint8)
        im=Image.fromarray(arr)
    
        palette=[]
        for i in range(256):
            palette.extend((i,i,i))
        palette[:3*21]=np.array([[0, 0, 0],
                                    [128, 0, 0],
                                    [0, 128, 0],
                                    [128, 128, 0],
                                    [0, 0, 128],
                                    [128, 0, 128],
                                    [0, 128, 128],
                                    [128, 128, 128],
                                    [64, 0, 0],
                                    [192, 0, 0],
                                    [64, 128, 0],
                                    [192, 128, 0],
                                    [64, 0, 128],
                                    [192, 0, 128],
                                    [64, 128, 128],
                                    [192, 128, 128],
                                    [0, 64, 0],
                                    [128, 64, 0],
                                    [0, 192, 0],
                                    [128, 192, 0],
                                    [0, 64, 128]
                                 ], dtype='uint8').flatten()
    
        im.putpalette(palette)
        im.save(name)

参考：

[https://www.zhihu.com/question/263994588][https_www.zhihu.com_question_263994588]

[https://zhuanlan.zhihu.com/p/22976342][https_zhuanlan.zhihu.com_p_22976342]

[https_www.zhihu.com_question_263994588]: https://www.zhihu.com/question/263994588
[https_zhuanlan.zhihu.com_p_22976342]: https://zhuanlan.zhihu.com/p/22976342