本文代码是参考github上的simple-faster-rcnn-pytorch:https://github.com/zylhub/simple-faster-rcnn-pytorch
由于代码原作者能力太强 ,我太弱了,很多看不懂,所有先提前把使用放上来供大家参考。
修改2个地方就可以使用自己的数据集,不过数据集需要制作成voc2007里面的格式。
1.将voc_dataset.py和vis_tool.py里面的VOC_BBOX_LABEL_NAMES类别替换成自己标签中类别名称,必须和xml文件里面 的标签名字一致,否则会直接跳过这个样本。
2.修改config.py中的config.py中的voc_data_dir路径,改成自己的路径.
3.最后更改faster_rcnn_vgg16.py 中的n_fg_class参数值,来调整最后一层(其实自己的样本只要小于21类这个地方可以不改)
运行程序后可以打开http://localhost:8097/# 查看程序的中间结果
这个是visdom可视化,所有必须先运行指令:
python -m visdom.server
have fun!