pytorch 从头开始faster-rcnn 最后：使用自己的数据集

本文代码是参考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！