本片文章借鉴以下代码
源码地址:https://github.com/bubbliiiing/faster-rcnn-pytorch
本片文章借鉴以下博客
博客地址:https://blog.csdn.net/weixin_44791964/article/details/105739918
本代码使用VOC格式进行训练。
训练前将标签文件放在VOCdevkit文件夹下的VOC2007文件夹下的Annotation中,
将图片文件放在VOCdevkit文件夹下的VOC2007文件夹下的JPEGImages中。
VOC下载 https://pjreddie.com/projects/pascal-voc-dataset-mirror/
(建议先处理一下,用VOC官网代码可能会使电脑炸掉)
Faster-rcnn文件架构如下 
1、训练前利用voc2faster-rcnn.py文件生成对应的txt

这一步一般不会出错,有可能报错的便是下面的文件路径,需要根据自己Annotation和JPEGImages文件的的位置设置文件路径
再运行根目录下的voc_annotation.py,运行前需要将classes改成你自己的classes。
同时在训练前需要修改model_data里面的voc_classes.txt文件,需要将classes改成你自己的classes。
将train.py文件下的NUM_CLASSES修改成需要分的类的个数。
运行train.py即可开始训练
(做好心理准备,要运行好长时间大概1天半左右)
这两个文件需要自己下载好voc-weights-resnet.pth
voc-weights-resnet.pth

自行修改你你训练时要用的网络
在代码运行过程中如果发现部分库未安装自行下载
再次声明本文主要借鉴于以下博客
博客地址:https://blog.csdn.net/weixin_44791964/article/details/105739918
