mmdetection使用Faster-RCNN训练自己的coco格式数据集
6.在mmdet/datasets/coco.py中修改自己的数据集类别名称，在mmdet/evaluation/functional/class_names.py中修改自己的类别。4.在mmdet/configs/_base_/datasets/coco_detection.py中修改自己的数据集路径和训练图片的imgsz。3.进入configs/_base_/models/faster-rcn
参考博客使用MMDetection训练自己的数据集(COCO)-CSDN博客
1.准备好数据集
2.在主目录的configs/faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py找到需要配置的文件，依次修改
3.进入configs/_base_/models/faster-rcnn_r50_fpn.py，修改num_classes为自己的类别数
4.在mmdet/configs/_base_/datasets/coco_detection.py中修改自己的数据集路径和训练图片的imgsz
5.在使用的mmdet/configs/_base_/schedules/schedule_1x.py中修改自己的epoch等
6.在mmdet/datasets/coco.py中修改自己的数据集类别名称，在mmdet/evaluation/functional/class_names.py中修改自己的类别
7.运行命令
python tools/train.py E:/GoogleDownload/mmdetection-main/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py --work-dir run_workstation
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
8
0- 0
已为社区贡献1条内容
所有评论(0)