https://zhuanlan.zhihu.com/p/162730118
https://blog.csdn.net/weixin_42146720/article/details/114543889
https://www.it610.com/article/1294190526756626432.htm

前言

MMdetection2.X配置与使用COCO格式数据集的记录。

一、按照官方文档配置并测试demo

可以按照官方网站或者mmdetection-master中readme进行配置

二、将COCO格式的数据集划分为训练、验证集并放入data文件夹下

目录形式如下

三、修改配置文件

以cascade_rcnn_r50_fpn_1x_coco为例
通过configs文件夹，cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.py

其中models和datasets修改相应的数据集类别数num_classes,其他的也可按需修改，lr在schedules中。
注：2.x版本中num_classes不需要+1

四、修改mmdet/core/evalution/class_names.py和mmdet/datasets/coco.py中的标签

修改完后，需要重新编译（python setup.py install)，不然会出现“AssertionError: The num_classes (20) in Shared2FCBBoxHead of MMDataParallel does not matches the length of CLASSES 80) in RepeatDataset"的报错。

五、按照官方说明训练

版权声明：本文为CSDN博主「leozhang97」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_41690533/article/details/115868856