m0_51472552的博客

原创 使用Pytorch实现DenseNet

代码中的conv_block即为上图中绿色框的部分，先使用1*1卷积来减少通道数从而减少参数量，论文中使用4*k作为该次卷积的输出通道数（k代表每个conv_block输出的通道数，也就是论文中growth rate，是固定值）接着使用3*3的卷积，通道数为k。其中theta是论文中所使用的超参数，用theta*in_channel来表示输出的通道数，为了达到减少参数量的目的，theta取值0-1之间，论文中设为0.5。上图中表示为红色框住的部分，代码中即为transition类。