实例分割之SOLOv2: Dynamic, Faster and Stronger

论文

代码

SOLOv2为作者对SOLO的改进，采用动态卷积核生成实例掩码。若对SOLO不了解，可先戳这里。

 

作者取名为Dynamic head。将SOLO中的Mask Branch改为kernel branch和feature branch两个分支。

Dynamic head

• feature branch的特征图F的生成。E为特征图的通道数，论文给出的为256.

• kernel branch为卷积核生成，每个格子对应一个卷积核。其中D为卷积核的参数。如采用3*3的卷积核，那么D=3*3*E，对应的卷积核的shape为[1, E, 3, 3]，分别为输出的通道数、输入的通道数、kernel size。论文中，作者采用的卷积核shape为[1,E,1,1]，即作者采用了1*1的卷积核。
• 每个格子对应的实例分割图为该格子预测的卷积核与特征图F的卷积结果。

与SOLO相比，除了将SOLO中的Mask branch替换为Dynamic head外，其他地方未做明显改动。但是与SOLOv1相比，速度更快，精度更好。