1.介绍
DenseBlock是DenseNet网络的重要组成部分,下面图片展示的是从DenseNet网络中节选出来的DenseBlock模块。
其主要思想是:对于每一层,前面所有层的特征映射作为当前层输入,而自己的特征映射为后续层输入,形成全互链接。每层提取出来的特征映射都可以供后续层使用。
优点:1.缓解梯度消失;2.增强特征传播;3.降低参数数量。
2.代码一
class _DenseLayer(nn.Sequential):
'''DenseBlock中的内部结构,这里是BN+ReLU+1x1 Conv+BN+ReLU+3x3 Conv结构'''
def