search

HuCaoFighting / Swin-Unet

[ECCVW 2022] The codes for the work "Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentation"
1.58k stars 298 forks source link

skip connection的问题 #66

Open stdcoutzrh opened 2 years ago

stdcoutzrh commented 2 years ago

请问是不是应该把transformer block的输出跳跃连接到后面，作者的代码似乎是把tfm block之前的patch merging下采样的输出给skip到解码器的。

HuCaoFighting commented 2 years ago

可以把不同的输出skip connection到decoder。这个不影响整体架构。

rocklijun commented 1 year ago

可以把不同的输出skip connection到decoder。这个不影响整体架构。

作者你好，但是论文里画的框架却是把transformer block 的输出跳跃连接到decoder，这和代码有点出入，改一下应该更好吧。