当前位置: 首页 > 网络应用技术

使用MMDetection训练您的数据集(VOC格式)

时间:2023-03-06 00:24:18 网络应用技术

  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。如果要使用其他模型配置文件,则可以自己修改。

  生成的文件如下:

  要修改的地点包括类别,数据集,数据集目录,日志目录和评估标准的数量。特定的修改如下