计算机视觉系列-轻松掌握 MMDetection 中常用算法:Faster R-CNN|Mask R-CNN (三)

计算机视觉系列-轻松掌握 MMDetection 中常用算法:Faster R-CNN|Mask R-CNN (三)

Faster R-CNN 代码详解

BBox Encoder Decoder

在 anchor-based 算法中,为了利用 anchor 信息进行更快更好的收敛,一般会对 head 输出的 bbox 分支 4 个值进行编解码操作,作用有两个:

  • 更好的平衡分类和回归分支 loss,以及平衡 bbox 四个预测值的 loss
  • 训练过程中引入 anchor 信息,加快收敛

Faster R-CNN中BBox Encoder Decoder参数及实现方式和 RetinaNet 相同

采用的编解码函数是 DeltaXYWHBBoxCoder,其配置如下

            bbox_coder

猜你喜欢

转载自blog.csdn.net/duan_zhihua/article/details/125187601