mmdetection在SSD网络上训练自己的COCO数据集
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为自己数据的东西,且要修改模型输出分类数量。