计算机视觉系列-轻松掌握 MMDetection 中常用算法 ：RetinaNet（三）

计算机视觉系列-轻松掌握 MMDetection 中常用算法 ：RetinaNet（三）

RetinaNet 代码详解 Head

• RetinaNet 的 Head 模块比较重量级，输出头包括分类和检测两个分支，且每个分支都包括 4 个卷积层，不进行参数共享
• 分类 Head 输出通道是 num_class*K
• 检测 head 输出通道是4*K
• K 是 anchor 个数, 虽然每个 Head 的分类和回归分支权重不共享，但是 5 个输出特征图的 Head 模块权重是共享的。

Head的配置：

 bbox_head=dict(
        type