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

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

Faster R-CNN 代码详解

RoI Head

R-CNN 模块( RoI Head)接收 RPN 输出的每张图片共 nms_post 个候选框,refine输出包括区分具体类别和 bbox 回归,包括了 RPN 中涉及到的组件,例如 BBox Assigner、BBox Sampler、BBox Encoder Decoder、Loss 等等,还包括一个额外的 RPN 到 R-CNN 数据转换模块:RoIAlign 或者 RoIPool

    roi_head=dict(
        type='StandardRoIHead',
        bbox_roi_extractor=dict(
            type

猜你喜欢

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