计算机视觉系列-轻松掌握 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 分支