此外,在parse_dataset函数中,保存了类别名到id的映射的一个字典 cname2cid。在coco数据集中,会利用 COCO API 从标注文件中加载数据集的类别名,并设置此字典。在voc数据集中,如果设置 use_default_label=False,将从 label_list.txt 中读取类别列表,反之将使用voc默认的类别列表。
在 source/coco.py 中定义并注册了 COCODataSet 数据集类,其继承自 DetDataSet,并实现了parse_dataset方法,调用 COCO API 加载并解析COCO格式数据源 roidbs 和 cname2cid,具体可参见 source/coco.py 源码。将其他数据集转换成COCO格式可以参考 用户数据转成COCO数据 ...