simple-faster-rcnn-pytorch 训练自己的数据 (pascal voc格式)

项目地址:simple-faster-rcnn-pytorch

修改脚本:

  • 将voc_dataset.py和vis_tool.py里面的VOC_BBOX_LABEL_NAMES类别替换成自己标签中类别名称,必须和xml文件里面的标签名字一致,否则会直接跳过这个样本。
  • 修改config.py中的config.py中的voc_data_dir路径,改成自己的路径.
  • 修改faster_rcnn_vgg16.py  中的n_fg_class参数值,来调整最后一层(其实自己的样本只要小于21类这个地方可以不改)
  • 修改trainer.py中 self.roi_cm = ConfusionMeter(21) 将21改成自己标签中类别名+1

参考链接

你可能感兴趣的:(pytorch,faster-rcnn)