不明白原理的移步:这里
代码
文件夹结构
faster RCNN
|-data
|-default
|-lib
|-demo.py
|-train.py
各文件夹详细介绍
data:数据集的读取与候选框的相关操作
default:记录训练过程产生的参数
lib:整个faster RCNN的网络结构
config:配置文件,config.py对参数进行设置,包括训练次数,锚框的最大值最小值,路径等
datasets:训练集读取,包括图片读取和锚宽坐标等信息
layer_utils:锚框与推荐区域尺寸设置
nets:网络结构,其中network.py是proposal和rpn构建函数;vgg16.py为调用了network.py并添加了vgg16的网络结构;resnet.py为调用了network.py并添加了resnet的网络结构
utils: