本文相关
-
Paper Summary
:https://github.com/FDU-VTS/CVPaper -
Code
:https://github.com/FDU-VTS/CVCode
原文链接
定义
- region proposal(region of interest): 候选区域
- ground truth: 样本区域
- IoU: 计算重叠面积 = (A∩B) / (A∪B)
算法流程
CNN训练
- 数据集准备
- 选取训练集中的
ground truth
,并使用selective search
计算出region proposals
- 计算
ground truth
和roi
之间的IoU
,选取其中值大于0.5的roi
s,然后选取25%计入正样本中 - 第二步中
IoU
小于0.5中选取正样本3倍的数量记为background
- 选取训练集中的
- 训练网络
-
alex_net
/VGG-16
-
- 损失函数
- cross entropy
SVM训练
- 数据集准备
- 按照CNN训练中的,不过这次挑选
IoU
大于0.5的proposals
, 和ground truth
一起组成正样本集合 -
IoU
小于0.3的归为负样本集合
- 按照CNN训练中的,不过这次挑选
- 训练m个类别的SVM
bounding box regression
- 坐标和尺度变换:
- 损失函数
- 正确值
网友评论