cityscapes数据集
name id trainId category catId hasInstances ignoreInEval color
--------------------------------------------------------------------------------------------------------------------
unlabeled' , 0 , 255 , 'void' , 0 , False , True , ( 0, 0, 0) ),
ego vehicle' , 1 , 255 , 'void' , 0 , False , True , ( 0, 0, 0) ),
rectification border' , 2 , 255 , 'void' , 0 , False , True , ( 0, 0, 0) ),
out of roi' , 3 , 255 , 'void' , 0 , False , True , ( 0, 0, 0) ),
static' , 4 , 255 , 'void' , 0 , False , True , ( 0, 0, 0) ),
dynamic' , 5 , 255 , 'void' , 0 , False , True , (111, 74, 0) ),
ground' , 6 , 255 , 'void' , 0 , False , True , ( 81, 0, 81) ),
road' , 7 , 0 , 'ground' , 1 , False , False , (128, 64,128) ),
sidewalk' , 8 , 1 , 'ground' , 1 , False , False , (244, 35,232) ),
parking' , 9 , 255 , 'ground' , 1 , False , True , (250,170,160) ),
rail track' , 10 , 255 , 'ground' , 1 , False , True , (230,150,140) ),
building' , 11 , 2 , 'construction' , 2 , False , False , ( 70, 70, 70) ),
wall' , 12 , 3 , 'construction' , 2 , False , False , (102,102,156) ),
fence' , 13 , 4 , 'construction' , 2 , False , False , (190,153,153) ),
guard rail' , 14 , 255 , 'construction' , 2 , False , True , (180,165,180) ),
bridge' , 15 , 255 , 'construction' , 2 , False , True , (150,100,100) ),
tunnel' , 16 , 255 , 'construction' , 2 , False , True , (150,120, 90) ),
pole' , 17 , 5 , 'object' , 3 , False , False , (153,153,153) ),
polegroup' , 18 , 255 , 'object' , 3 , False , True , (153,153,153) ),
traffic light' , 19 , 6 , 'object' , 3 , False , False , (250,170, 30) ),
traffic sign' , 20 , 7 , 'object' , 3 , False , False , (220,220, 0) ),
vegetation' , 21 , 8 , 'nature' , 4 , False , False , (107,142, 35) ),
terrain' , 22 , 9 , 'nature' , 4 , False , False , (152,251,152) ),
sky' , 23 , 10 , 'sky' , 5 , False , False , ( 70,130,180) ),
person' , 24 , 11 , 'human' , 6 , True , False , (220, 20, 60) ),
rider' , 25 , 12 , 'human' , 6 , True , False , (255, 0, 0) ),
car' , 26 , 13 , 'vehicle' , 7 , True , False , ( 0, 0,142) ),
truck' , 27 , 14 , 'vehicle' , 7 , True , False , ( 0, 0, 70) ),
bus' , 28 , 15 , 'vehicle' , 7 , True , False , ( 0, 60,100) ),
caravan' , 29 , 255 , 'vehicle' , 7 , True , True , ( 0, 0, 90) ),
trailer' , 30 , 255 , 'vehicle' , 7 , True , True , ( 0, 0,110) ),
train' , 31 , 16 , 'vehicle' , 7 , True , False , ( 0, 80,100) ),
motorcycle' , 32 , 17 , 'vehicle' , 7 , True , False , ( 0, 0,230) ),
bicycle' , 33 , 18 , 'vehicle' , 7 , True , False , (119, 11, 32) ),
license plate' , -1 , -1 , 'vehicle' , 7 , False , True , ( 0, 0,142) ),
总共5000张精细释,2975张训练图,500张验证图和1525张测试图。
原始精细标注数据集里面其实每张图片只对应四张标注文件:
xxx_gtFine_color.png, : 标注的可视化图片
xxx_gtFine_instanceIds.png :是用来做实例分割训练用的
xxx_gtFine_labelsIds.png :是语义分割训练需要的,它们的像素值就是class值
xxx_gtFine_polygons.json :用labelme工具标注后所生成的文件,里面主要记录了每个多边形标注框上的点集坐标。
另外两个
xxx_gtFine_instanceTrainIds.png
xxx_gtFine_labelTrainIds.png
使用labels.py (from https://github.com/mcordts/cityscapesScripts/tree/master/cityscapesscripts/helpers)来生成的。
因为实际上这5000张精细标注的图片有34类(0~33),
但训练时可能只想关心其中19类(0~18)。
所以需要做一个映射来将34类中感兴趣的类别映射到19类中,
其它不感兴趣的类别就直接设成255,所以这也是为什么xxx_trainIds.png中有白色像素的原因,因为那些白色像素的类别不是我们感兴趣的,变成255白色了。