【mmdetection训练自己的数据集COCO】faster-rcnn

1.官网下载mmdetection源码https://github.com/open-mmlab/mmdetection

2.读中文文档配置好相关环境

3.定义数据种类，打开mmdetection-main/mmdet/datasets/coco.py，将classes中的数据改为自己数据集的种类即可。例如：

 4.接着在mmdetection-main/mmdet/evaluation/functional/class_names.py中修改coco_classes数据集类别，例如：

5.打开配置文件mmdetection-main/configs/faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py，修改相关参数，必改 num_classes类别，data_root数据集位置。

6.如果需显示每一类AP值，更改mmdetectionmain/mmdet/evaluation/metrics/coco_metric.py中的classwise为True即可。

7.单GPU开始训练，在终端输入python tools/train.py configs/faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py即可。