当前位置:   article > 正文

【目标检测】使用MMDetection训练自定义COCO格式数据集_mmdetection之queryins 训练coco

mmdetection之queryins 训练coco

1.自定义数据集(COCO格式)的准备

1.1. 将数据集按照以下路径存放到MMDet目录

  1. mmdetection
  2. |--data
  3. | |--coco
  4. | |--annotations
  5. | |--instances_train2017.json
  6. | |--instances_val2017.json
  7. |
  8. | |--train2017
  9. | |--1.jpg
  10. | |--2.jpg
  11. | |--...
  12. |
  13. | |--val2017
  14. | |--7.jpg
  15. | |--26.jpg
  16. | |--...

1.2. 选择模型和预训练权重

1.3. 修改数据集配置文件

查看所选模型的配置文件,以训练Faster R-CNN为例:

首先修改COCO数据集配置文件:configs/_base_/datasets/coco_detection.py

添加自定义数据集的类别信息

1.4. 修改模型配置文件的类别信息

模型配置文件路径:configs/_base_/models/faster_rcnn_r50_fpn.py

将类别数从80修改为自定义数据集的类别数: 

1.5. 设置预训练权重

配置文件路径:configs/_base_/default_runtime.py

添加预训练权重路径: 

1.6. 每次修改完config文件需进行setup

python setup.py install

2. 模型训练

python tools/train.py config.py

以训练Faster R-CNN为例:

python tools/train.py configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py

MMDet常用指令:

【目标检测】MMDetection常用指令_ericdiii的博客-CSDN博客目标检测MMDetection库常用指令https://blog.csdn.net/ericdiii/article/details/125722277?spm=1001.2014.3001.5502

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/酷酷是懒虫/article/detail/762142
推荐阅读
相关标签
  

闽ICP备14008679号