mmdetection[v1.1.0]训练自己的VOC数据集

之前写过一个版本,现在不怎么适用了。所以重新写下。

步骤一:首先根据规范的VOC数据集导入到项目目录下,如下图所示:

mmdetection
├── mmdet
├── tools
├── configs
├── data
│   ├── VOCdevkit
│   │   ├── VOC2007
│   │   │   ├── Annotations
│   │   │   ├── JPEGImages
│   │   │   ├── ImageSets
│   │   │   │   ├── Main
│   │   │   │   │   ├── test.txt
│   │   │   │   │   ├── trainval.txt
然后复制configs/fast_rcnn_r50_fpn_1x.py ,更名为my.py

步骤二:修改my.py, 

mmdetection[v1.1.0]训练自己的VOC数据集

上图中num_classes=自己的类数+1

再修改下图中4个地方

mmdetection[v1.1.0]训练自己的VOC数据集

步骤三:修改mmdetection/mmdet/datasets目录下voc.py

mmdetection[v1.1.0]训练自己的VOC数据集

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

mmdetection[v1.1.0]训练自己的VOC数据集

运行python setup.py install,重新编译

运行代码 python ./tools/train.py configs/my.py --work_dir model --validate

周一再补