神经网络改进常用模块(需要一定的修改)

本文探讨了神经网络中的几个关键模块,包括循环卷积、残差模块及其结合形式残差循环卷积,深入研究了3*3卷积和1*1卷积在特征处理中的作用,尤其是1*1卷积在降低维度方面的效果,同时提到了密集链接模块在复杂网络架构中的应用。
摘要由CSDN通过智能技术生成

循环卷积模块

class Recurrent_block(nn.Module):
    """
    Recurrent Block for R2Unet_CNN
    *********************************************循环卷积模块*********************************************
    """
    def __init__(self, out_ch, t=2):
        super(Recurrent_block, self).__init__()

        self.t = t
        self.out_ch = out_ch
        self.conv = nn.Sequential(
            nn.Conv2d(out_ch, out_ch, kernel_size=3, stride=1, padding=1, bias=True),
            nn.BatchNorm2d(out_ch),
            nn.ReLU(inplace=True)
        )

    def forward(self, x):
        for i in range(self.t):
            if i == 0:
                x = self.conv(x)
            out = self.conv(x + x)
        return out

残差模块

class Bottleneck(nn.Module):
    expansion = 4
    def __init__(self, inplanes, planes
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值