网站建设原创文章 关键词朋友圈广告代理商官网

mmdetection目录下创建data文件夹，路劲如图所示，不带yololabels

修改配置文件

mmdet/datasets/voc.py

配置图片格式

mmdet/datasets/xml_style.py

如果图片是jpg则改成jpg，是png格式就改成png，这里我不需要改，本身就是jpg

用自己的VOC数据集训练mmdetection中的faster_rcnn

faster_rcnn 基本配置文件位置：configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py

依次修改这几个配置文件

‘…/base/models/faster_rcnn_r50_fpn.py’, ‘…/base/datasets/voc0712.py’
…/base/models/faster_rcnn_r50_fpn.py修改类别数目

…/base/datasets/voc0712.py修改路径

建立文件夹的时候就是按照这个建立的，所以root路径不用改

修改mmdetection/mmdet/core/evaluation目录下class_names.py

然后运行！！！！！！！！！！！！！！！！！

python setup.py install

训练faster_rcnn

python tools/train.py configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py

训练ssd

同理修改配置文件
configs/base/models/ssd300.py

configs/pascal_voc/ssd300_voc0712.py

python tools/train.py configs/pascal_voc/ssd300_voc0712.py