0 mmdetection环境配置
GitHub官网

一、数据集准备工作
数据集目录格式要求：
图片说明

PS：

  1. 所有图片末尾统一为.jpg（注意小写）
    数据集保存（创建软连接，方便数据集复用）： 
mkdir data
cd data
ln -s /home/存访路径/VOCdevkit

二、修改相关文件

1. 修改class_names.py文件
修改路径:

mmdetection/mmdet/core/evaluation/class_names.py

修改内容：将voc_classes的返回值改为要训练数据集的类别名称。

图片说明

2. 修改voc.py文件
修改路径：

mmdetection/mmdet/datasets/voc.py

修改内容：将VOCDataset中的CLASSES改为对应训练数据集的类别集合。

图片说明

图片说明

1. models中对应文件修改num_classes变量

图片说明

2. datasets中对应文件修改dataset_type、data_root、img_scale、ann_file、img_prefix变量。

图片说明

图片说明

3. schedules中新建schedule_xxx_voc.py设置对应优化器

图片说明

4. 新建xxx_voc_runtime.py文件设置整体超参

图片说明

三、开始训练

python tools/train.py configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_voc_xxx.py

对应log日志以及断点保存在work_dirs目录下。

Reference

  1. https://blog.csdn.net/laizi_laizi/article/details/104256781

  2. https://blog.csdn.net/qq_43271903/article/details/112554903?utm_medium=distribute.pc_relevant.none-task-blog-baidujs_title-0&spm=1001.2101.3001.4242

  3. https://blog.csdn.net/qq_30121457/article/details/112591430?utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-4.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-4.control
    ————————————————
    版权声明：本文为CSDN博主「阿基美德」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
    原文链接：https://blog.csdn.net/X_blackbutterfly/article/details/118256836