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,
上图中num_classes=自己的类数+1
再修改下图中4个地方
步骤三:修改mmdetection/mmdet/datasets目录下voc.py
步骤:修改mmdetection/mmdet/core/evaluation目录下class_names.py
运行python setup.py install,重新编译
运行代码 python ./tools/train.py configs/my.py --work_dir model --validate
周一再补