U2-NET
改进
网络层面的改进,提出了一个RSU结构,类似于一个小型的unet结构。
U2-NET对于分割物体前背景取得了惊人的效果。同样具有较好的实时性,经过测试在P100上前向时间仅为18ms(56fps)。
RSU结构(ReSidual U-blocks)

RSU结构类似于一个小型的U-Net,将所有的RSU用一种类似U-Net的结构连接在一起。通过这种方式来增加多尺度能力。获得了极为优秀的分割结果。
整体网络

类似U-Net的网络,将其内部换成了RSU结构,每一层输出一个mask,最后将mask进行叠加经过卷积得到最终的mask。
loss函数
由七个loss相加(一共六层,每一层一个loss。再加上最后mask融合时的一个loss)。
6516

被折叠的 条评论
为什么被折叠?



