文章目录[隐藏]
数据集存放位置与格式
├── coco
│ ├── annotations
│ ├── test2017
│ ├── train2017
│ └── val2017
更改的配置文件
./configs/_base_/default_runtime.py
:决定是否启用tensorboard
/mmdet/datasets/coco.py
:把CLASSES的那个tuple改为自己数据集对应的种类tuple。
CLASSES = ('people', 'bicycle', 'car', 'van', 'truck', 'tricycle', 'bus', 'motor','boat')
./mmdet/core/evaluation/class_names.py
:找到def coco_classes()
更改为自己的数据集名称
return ['people', 'bicycle', 'car', 'van', 'truck', 'tricycle', 'bus', 'motor','boat']
版权声明:本文为CSDN博主「是王同学呀」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/wxd1233/article/details/123030468
暂无评论