计算机视觉系列-轻松掌握 MMDetection 中常用算法 ：Cascade R-CNN（二）

Cascade R-CNN 代码实现

Cascade R-CNN 的Backbone、FPN 和 RPN
这三个部分和 Faster R-CNN 基本相同，本文从上层对比 Faster R-CNN 和 Cascade R-CNN 的 R-CNN 训练和测试的区别

• Faster R-CNN 的 StandardRoIHead
  mmdetection-2.25.0/mmdet/models/roi_heads/standard_roi_head.py

• Cascade R-CNN的CascadeRoIHead
  mmdetection-2.25.0/mmdet/models/roi_heads/cascade_roi_head.py

Faster R-CNN 训练流程

暂时不考虑 mask 场景

    def forward_train(self,
                      x,
                      img_metas,
                      proposal_list
1
2
3