unet简单改进版_改进uent创新点代码

作者：Monodyee | 2024-03-17 15:14:52

踩

改进uent创新点代码

U-Net网络模型（简单改进版）
这一段时间做项目用到了U-Net网络模型，但是原始的U-Net网络还有很大的改良空间，随手加了点东西：每次的下采样的过程中加入了通道注意力和空间注意力。
代码跑出来后，效果比原来的U-Net大概提升了一个点左右，证明是有效的，改动很少，放出代码：
在这里插入图片描述

class ChannelAttentionModule(nn.Module):
    def __init__(self, channel, ratio=16)1

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Monodyee/article/detail/256737