本片文章借鉴以下代码
    源码地址: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文件架构如下
    image.png

    1、训练前利用voc2faster-rcnn.py文件生成对应的txt

    image.png

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

    这两个文件需要自己下载好voc-weights-resnet.pth
    voc-weights-resnet.pth

    image.png

    自行修改你你训练时要用的网络
    image.png

    在代码运行过程中如果发现部分库未安装自行下载

    再次声明本文主要借鉴于以下博客
    博客地址:https://blog.csdn.net/weixin_44791964/article/details/105739918