【YOLACT】代码解读

源码:https://github.com/dbolya/yolact

YOLACT的backbone结构

 其中_make_layer就是常规的resnet101中的_make_layer,具体是如何操作的可以查看源码https://github.com/dbolya/yolact

 

【YOLACT】代码解读

其中浅蓝色的就是后续将要使用的特征图。

YOLACT的FPN结构

【YOLACT】代码解读

FPN选用了backbone结构结构中的1、2、3作为输入,得到新的输出,也就是橘黄色的三角形0到4。

YOLACT的proto结构

【YOLACT】代码解读

将FPN中的输出0作为proto的输入,最终得到proto的输出(1,138,138,32)。

YOLACT的pred_heads结构

【YOLACT】代码解读

其中的3表示每个位置有3个预设anchor_box,然后对FPN中的所有特征图都进行操作,得到对应的bbox,conf,mask,priors,proto。

还有一个输出segm,具体的操作如下图所示:

【YOLACT】代码解读