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

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

Faster R-CNN 代码详解

Faster R-CNN 模型整体流程:

在这里插入图片描述

  • 图片输入到 ResNet 中进行特征提取,输出 4 个特征图,stride = 4,8,16,32
  • 4 个特征图输入到 FPN 模块中进行特征融合,输出 5 个通道数相同的特征图, stride = 4,8,16,32,64
  • FPN 输出的 5 个特征图,输入到同一个 RPN( 5 个相同的 RPN) 中,每个分支都进行前后景分类和 bbox 回归,然后和 label 计算 loss
  • 在 5 个 RPN 分支

猜你喜欢

转载自blog.csdn.net/duan_zhihua/article/details/125171571
今日推荐