计算机视觉系列-轻松掌握 MMDetection 中常用算法:Faster R-CNN|Mask R-CNN (三)
目录
Faster R-CNN 代码详解
BBox Encoder Decoder
- Faster R-CNN 代码详解
-
- BBox Encoder Decoder
- Loss
- RPN Test
- RoI Head
- 计算机视觉学习笔记系列
在 anchor-based 算法中,为了利用 anchor 信息进行更快更好的收敛,一般会对 head 输出的 bbox 分支 4 个值进行编解码操作,作用有两个:
- 更好的平衡分类和回归分支 loss,以及平衡 bbox 四个预测值的 loss
- 训练过程中引入 anchor 信息,加快收敛
Faster R-CNN中BBox Encoder Decoder参数及实现方式和 RetinaNet 相同
采用的编解码函数是 DeltaXYWHBBoxCoder,其配置如下
bbox_coder