torch.ConvTranspose2d的计算

转置卷积实际是根据输入卷积核和输入特征图生成中间卷积核和特征图,再进行卷积操作

输入的卷积核kernel 大小 Size = size 步长 Stride = stride 填充 Padding = padding

输入特征图 H W

中间卷积核kernel' 大小 Size' = size 步长 Stride' = 1 填充 Padding' = size - padding - 1

中间特征图 H' = H + (stride-1) * (H-1)

输出特征图 H'' = (H' + 2*Padding' - Size') / Stride' +1

         = stride (H - 1) + size -2*padding

posted @ 2020-03-16 10:54  Junzhao  阅读(771)  评论(0编辑  收藏  举报