Pytorch版Faster R-CNN 源码分析+方法流程详解——训练篇
继demo篇之后,继续分析faster rcnn的训练过程,继续对Pytorch版的源码进行分析。
1、参考文章/博客/论文/源码
论文:https://arxiv.org/pdf/1506.01497.pdf
源码:https://github.com/jwyang/faster-rcnn.pytorch
2、环境配置
见demo篇faster rcnn检测阶段源码分析
3、数据集及模型
以VGG16作为Backbone网络模型,数据集采用PASCAL VOC2007。
5、faster rcnn方法训练流程分析(训练阶段)
脚本文件:trainval_net.py
1、def parse_args():
参数传递函数,定义网络训练所需的相关参数,方便使用命令传递参数。