《深入理解并实现Att U-Net》

本文介绍了《Attention U-Net: Learning Where to Look for the Pancreas》中提出的Att U-Net网络,它通过Attention Gate学习关键特征进行胰腺分割。网络结构基于U-Net,但加入了Attention Gate来加权编码器信息,提高分割精度。训练结果显示了网络的有效性。
摘要由CSDN通过智能技术生成

Introduction

       《Attention U-Net: Learning Where to Look for the Pancreas》发表于2018CVPR,相对于定位+分割的级联方式实现的精准语义分割,通过添加attention gate的方式隐式的学习对于胰腺分割重要的特征,省略不重要的特征。

Network Architecture

       网络结构类似于U-net,区别是U-net在解码器和编码器同层级只进行拼接操作,此处则加入了attention gate,用于将同层级的encoder部分编码进行加权;

 

Attention gate

Attention的计算过程为:首先将decoder部分对应的输出g进行上采样+卷积(作用和反卷积类似),加卷积的目的是使两部分的channel数一致;然后分别使用1维卷积将g和x(来自于同层次的encoder)进行降维,channel数变为原来的1/2。然后两部分结果相加,经过激活函数,一维卷积将channel数降至1,然后经过sigmod函数,得到和x大小相同的1维的attention map,和原来的x做element-wise的乘法,得到加权后的向量;

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值