conda install pytorch == 1.6.0 torchvision == 0.7.0 cudatoolkit = 10.1 -c pytorch -y
pip install install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.6.0/index.html
git克隆https://github.com/open-mmlab/mmdetection.git cd mmdetection pip install -r researchs/build.txt pip install -v -e。
1.修改mmdetection/configs/pascal_voc/faster_rcnn_r50_1x_voc0712.py作为培训的数据集数量
2.在num中修改num_classeseses
在源代码中给出的VOC格式数据集中的模型配置文件很少,仅FALCNN_R50,SSD。如果要使用其他模型配置文件,则可以自己修改。
生成的文件如下:
要修改的地点包括类别,数据集,数据集目录,日志目录和评估标准的数量。特定的修改如下