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
训练
python tools/train.py configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py
python tools/train.py configs/pascal_voc/ssd512_voc.py