Attention UNet的原理对应论文 Attention UNet:attention unet learning where to find the pancer

Attention Unet的理解
请参考这个视频(打开字幕):https://www.youtube.com/watch?v=KOF38xAvo8I

Attention UNet的原理对应论文 Attention UNet:attention unet learning where to find the pancer_第1张图片
解释:
这张图详细地描述了Attention UNet的结构。如图所示。x是指跳连接,x可以保留从低层传过来的spatial information(图中褐色框表示),g是指从底层传上来的信息(图中蓝框表示),g可以保留较多的feature representation。这里的Attention Gate是 soft attention
当我们expand the attention Gate,整体结构图如下图所示。整体有两个分支,一个是x 一个是g。g是图中的gating signal,which is coming from one step lower,and x is coming from one stop higher
Attention UNet的原理对应论文 Attention UNet:attention unet learning where to find the pancer_第2张图片

最后一步是把计算得到权重值(weight values)和原图相乘。类似于SE Module
(关于SE module的讲解,大家可以听这个视频https://www.bilibili.com/video/BV1sr4y1s7aL/)具体的代码如下所示:

你可能感兴趣的:(科研日常,笔记,学习生活,人工智能,深度学习)