- 深度学习高手笔记(卷2):经典应用
- 刘岩(@大师兄)
- 229字
- 2024-09-05 16:45:37
第1章 双阶段检测
1.3.5 Fast R-CNN的训练细节
1.迁移学习
同R-CNN一样,Fast R-CNN同样使用ImageNet的数据对模型进行预训练。详细地讲,先使用1000类的ImageNet训练一个1000类的分类器,如图1.9的虚线部分所示;然后提取模型中的特征层及其以前的所有网络;最后使用Fast R-CNN的多任务模型训练网络,如图1.9的实线部分所示。
2.小批次训练
在Fast R-CNN中,设每个批次的大小是R。在抽样时,每次随机选择N幅图像,每幅图像中随机选择R/N个候选区域,在实验中N=2、R=128。对候选区域进行抽样时,选取25%的正样本(和真值框的IoU大于0.5),75%的负样本。