ResNet改进(37)：DenseBlock模块实现

1.代码详解

这段代码实现了一个基于ResNet34架构的改进模型，在原始ResNet34的基础上添加了一个DenseBlock模块。

下面我将详细解释代码的各个部分：

1. DenseBlock模块

class DenseBlock(nn.Module):def __init__(self, in_channels, growth_rate=32):super(DenseBlock, self).__init__()self.bn1 = nn.BatchNorm2d(in_channels)self.conv1 = nn.Conv2d(in_channels, growth_rate, kernel_size=3, padding=1)def forward(self, x):out = self.conv1(nn.functional.relu(self.bn1(x)))out = torch.cat([x, out], 1)  # 密集连接return out

这个模块实现了DenseNet的核心思想——密集连接&