残差网络连接,使得输入与输出的尺寸一样

   def forward(self, x):
        out = self.layer1(x)
        out = self.layer2(out)
        # 使用插值将输入x上采样至与layer2输出相同的尺寸
        x = F.interpolate(x, size=(out.size(2), out.size(3)), mode='bilinear', align_corners=False)
         # 确保x的通道数与out匹配
        x = x[:, :out.size(1), :, :]  # 选择前out.size(1)个通道
        out += x  # 残差连接,将输入直接添加到layer2的输出上
        out = out.reshape(out.size(0), -1)
        out = self.dropout(out)
        out = self.fc(out)
        return out

 

这里采用双线性插值法 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值