修改源码部分(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/