计算机视觉系列-轻松掌握 MMDetection 中 Head 流程-学习笔记 三

计算机视觉系列-轻松掌握 MMDetection 中 Head 流程-学习笔记 三 本文继续学习MMDetection Head中roi_heads 模块训练和测试流程的内容

roi_heads 模块训练和测试流程 roi_heads 训练流程

训练流程最外层 调用 forward_train： mmdetection-2.24.1/mmdet/models/roi_heads/standard_roi_head.py

以 StandardRoIHead 为例进行分析：

• StandardRoIHead 中 forward_train 函数仅仅是对内部的 bbox_head 相关函数进行调用，例如 get_targets 和 loss，本身 StandardRoIHead 类不做具体算法逻辑计算。

    def forward_train(self,
                      x,
                      img_metas,
                      proposal_list,
                      gt_bboxes