COCO数据集准备

将自己的数据改为COCO数据的格式

mmdetection
├── mmdet
├── tools
├── configs
├── data
│   ├── coco
│   │   ├── annotations
│   │   ├── train2017
│   │   ├── val2017
│   │   ├── test2017

修改mmdetection COCO文件

将下列文件中的CLASSES,coco_classes,num_classes修改为自己的数据集中的分类。

mmdetection/mmdet/datasets/coco.py
mmdetection/mmdet/evaluation/functional/class_names.py
configs/ssd_xxxx.py

COCO.py,class_names.py需要修改的是,数据中类的名字。 ssd_xxx.py(模型文件),需要修改的是,数据中类的数量。

对应为需要修改的是,你要识别的数据集中的类名。 修改模型匹配数据集,并匹配输出。

修改mmdetection中自带的COCOgategories为自己数据的东西,且要修改模型输出分类数量。

训练步骤和测试步骤与官方一致