LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.
2.68k stars 595 forks source link

about the Atten_block implementation compared with the paper #39

Closed XushengLee closed 5 years ago

XushengLee commented 5 years ago

according to the paper, HgWgDg is matched up with HxWxDx using trilinear interpolation, which is also called "Resampler" in the figure 2. but in the code of this repo, there is no this step, instead you use the feature map after the Up_conv step. So the code can be runned, But it is clearly different from what proposed in the figure 2. So I just wonder whether you implemented this way on purpose? And if yes, please enlighten me why it's better this way. BTW, thank you so much for this repo, it really helps me a lot. :)