Voc2012数据集分割怎么使用加载标注

比眉伴天荒 2022-01-25 14:44 361阅读 0赞

voc2007 与 voc2017是互斥的图片。这次说2012

解压统统使用tar -xf 使用命令行，会自动给你融合图片。

标注是图片，存储在/..../VOCdevkit/VOC2012/SegmentationClass

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70][]

标注一共是20类，加上背景21类，白边呢，算是未标注类，不计算到损失函数里，注意下。

读取代码：

import numpy as np
    from PIL import Image
    from matplotlib import pyplot as plt
    #标注地址
    vocdir = '/home/DATA/database/VOC2012/ts/'
    labdir = 'VOCdevkit/VOC2012/'
    task = 'SegmentationClass/'
    #数据set地址
    list_dir = '/home/DATA/database/VOC2012/ts/VOCdevkit/VOC2012/ImageSets/Segmentation/'
    list_name = 'train.txt'
    all_the_text = open(list_dir+list_name).read()
    train = all_the_text.split('\n')
    for ind in train[0:1]:
    #按照图片名字读取，实际上读取数据只需要这一句话。
        lab = Image.open(vocdir+labdir+task+ind+'.png')
    
        plt.imshow(lab)
        plt.savefig("readlab")

前面写了很多地址，真正的关键是读取使用Image.open以P通道读取，这样的话

>>uniq = np.unique(lab)
    >>uniq
    >>array([  0,   1,  15, 255], dtype=uint8)

这个255需要自己处理下，别跑网络的时候也计算到损失里面去

参考：

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

[https://arleyzhang.github.io/articles/1dc20586/][https_arleyzhang.github.io_articles_1dc20586]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70]: /images/20220125/c0e5614033064a368717b5ffc8e8e2dc.png
[https_www.zhihu.com_question_263994588]: https://www.zhihu.com/question/263994588
[https_arleyzhang.github.io_articles_1dc20586]: https://arleyzhang.github.io/articles/1dc20586/