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

计算机视觉系列-轻松掌握 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 测试流程

mmdetection-2.25.0\mmdet\models\roi_heads\test_mixins.py

     def simple_test_bboxes(self,
                           x,
                           img_metas,

猜你喜欢

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