这一段时间做项目用到了U-Net网络模型,但是原始的U-Net网络还有很大的改良空间,随手加了点东西: 每次的下采样的过程中加入了通道注意力和空间注意力(大概就是这样) 代码跑出来后,效果比原来的U-Net大概提升了一个点左右,证明是有效的,改动很少,放出代码 classChannelAttentionModule(nn.Module):def__init__(se...