You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

请求获取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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.25 06:34:29