mmdetection中的faster-rcnn训练自己的voc数据集
1、首先先安装好mmdetection
2、安装好后在mmdetection文件夹下新建一个data数据集把voc数据集放进去
3、然后找到你的目录下的config/faster-RCNN/faster_rcnn_r50_fpn_1x_voc0712.py（如果没有自己建一个），内容如下
_base_ = [
'../_base_/models/faster_rcnn_r50_fpn.py',
'../_base_/datasets/voc0712.py',
'../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
]
4、去到configs/base/models/faster_rcnn_r50_fpn.py找到你的num_class改为你自己的类别
5、去到configs/base/models/voc0712.py中改好路径
dataset_type = 'VOCDataset'
data_root = 'data/VOCdevkit/'
如果没有voc2012就注释掉
data_root + 'VOC2012/ImageSets/Main/trainval.txt'
然后去到mmdet/core/evaluation/class_names.py 将def voc_classes():改为自己的类，一个记得加个,在后面
def voc_classes():
return [
'自己的类',
]
6、最后输入
python tools/train.py configs/faster_rcnn/faster_rcnn_r50_fpn_1x_voc0712.py
开始训练自己的数据集
|