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