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