目标检测算法---faster rcnn 知识简要回顾(训练篇)

2022-05-10  本文已影响0人  教训小磊
Faster RCNN 的训练是一个交替训练的过程,主要是对RPN和Fast RCNN进行训练。 Faster RCNN训练流程

Faster RCNN训练策略

训练RPN网络

Faster RCNN的一大创新点就是提出了“RPN”网络,在提高精度的同时提高了速度,这里讲一下RPN网络的训练。
RPN网络训练有两个Loss:

pi表示网络预测出来第i个anchor是目标的概率,pi* 表示对应的Ground Truth。若第i个anchor与某个真实目标的iou大于0.7,则pi* =1,小于0.3,则pi* =0,其余情况不参与训练;
ti表示参数化坐标向量,表示预测框和anchor框之间的偏移,ti*表示对应的Ground Truth,表示GT框和anchor框之间的偏移。

smooth L1 loss

smooth L1 loss公式

smooth L1 loss结合了L1和L2损失的优点,相比L1损失,smooth L1在0点可导,不影响收敛;相比L2损失,smooth L1对离群点、异常值不敏感,梯度变化相对更小,训练时不容易跑飞,所以鲁棒性更强。

上一篇下一篇

猜你喜欢

热点阅读