mmdetection 训练数据

修改源码部分(voc数据集)以faster rcnn为例

1. config/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py

修改数目类别
下面展示一些 内联代码片。

model = dict(roi_head=dict(bbox_head=dict(num_classes=1)))

2.core/evaluation/class_names.py

修改检测类别名称

3. mmdet/datasets/voc.py

修改检测类别

注意：当目标类别为一类时需要检测类别为列表形式不是元组

4. python setup.py install

将检测文件复制到虚拟环境中。若上述文件发生变换需要重新执行，重新编译。

5.检测图像不为jpg时需要修改

修改datasets/xml_style.py 中的.jpg为.tiff

6.python setup.py install

将部分文件复制到虚拟环境中去。
/×××/anaconda3/envs/pytorch/lib/python3.6/site-packages/mmdet-2.7.0-py3.6.egg/mmdet/datasets/