请求获取mobilenetv2_coco_cityscapes_trainfine模型(DeepLab v3)的Cityscapes标签顺序
解决DeepLab v3 mobilenetv2_coco_cityscapes_trainfine模型的标签顺序问题
我之前也踩过这个坑!你用的mobilenetv2_coco_cityscapes_trainfine模型是基于COCO预训练权重在Cityscapes数据集上做fine-tune的,它的标签顺序并不是Cityscapes官网给出的全类别顺序——因为这个模型只保留了和COCO数据集重叠的类别,并且完全沿用了COCO的标签ID映射规则。
你需要的包含car、motorcycle(模型里标注为motorbike)的正确标签顺序数组如下:
LABEL_NAMES = [ 'background', 'aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor' ]
补充说明:
- 这个模型总共包含21个类别(含背景),都是COCO和Cityscapes的重叠类别,所以Cityscapes官网的全30+类别顺序自然和它不匹配。
- 你可以快速验证:用模型预测一张包含车辆的测试图,提取mask的数值ID,比如
car对应的是ID 7,motorbike对应的是ID 14,对照你的标注结果就能确认是否匹配。
内容的提问来源于stack exchange,提问作者Marius Hofmann
相关产品推荐
相关产品推荐

