项目地址: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