转置卷积(反卷积)

torch.nn.ConvTranspose2d(in_channels, out_channels, kernel_size, stride=1, padding=0, output_padding=0, groups=1, 
                         bias=True, dilation=1, padding_mode='zeros', device=None, dtype=None)

计算公式:

output = ( input - 1 ) * s + k - 2p

其中:output 是输出尺寸,input 是输入尺寸 , s是步长,k是卷积后大小,p是padding 

如果希望特征图尺寸变为原来2倍,通道变为原来一半:

torch.nn.ConvTranspose2d(in_channels, in_channels // 2, 2, 2)

你可能感兴趣的:(python,深度学习,计算机视觉)