current position:Home>The principle of Attention UNet corresponds to the paper Attention UNet: attention unet learning where to find the dancer

The principle of Attention UNet corresponds to the paper Attention UNet: attention unet learning where to find the dancer

2022-11-24 23:07:45Polder

Understanding of Attention Unet
Please refer to this video (with subtitles on): https://www.youtube.com/watch?v=KOF38xAvo8I

Insert image description here
Explanation:
This picture describes in detailThe structure of Attention UNet.as the picture shows.x refers to the jump connection, x can retain the spatial information transmitted from the corresponding layer (indicated by the light purple box in the figure), g refers to the information transmitted from the bottom layer (indicated by the blue box in the figure), g can retain more featuresrepresentation.
When we expand the Attention Gate, the overall structure diagram is shown in the figure below.
Insert picture description here
The whole has two branches, one is x and the other isg.
g is the gating signal in the figure, which is coming from one step lower, and x is coming from one stop higher
Insert image description here
The last step is to multiply the calculated weight values ​​with the original image.Similar to SE Module
(For the explanation of SE module, you can listen to this video https://www.bilibili.com/video/BV1sr4y1s7aL/) The specific code is as follows:
insert image description here
Insert picture description here

copyright notice
author[Polder],Please bring the original link to reprint, thank you.
https://en.chowdera.com/2022/328/202211242304480104.html

Random recommended