美文网首页
torch ConvTranspose2d

torch ConvTranspose2d

作者: 小姐姐催我改备注 | 来源:发表于2019-09-26 11:35 被阅读0次

1.torch 中ConvTranspose2d

是卷积的一个逆过程,这里来研究下,怎么设置kenel_size,stride,padding,
首先给出卷积的计算公式:

output = (input - kenel + 2 * padding) / stride + 1
exm:
(112 - 3 + 2* 1)/2 + 1 = 56
这里是减半的核,
那么反卷积该怎么凑这三个参数呢。
这里分三种情况:
1.步长为1有填充: o = i + (k-1) - 2p 
    self.g1 = nn.ConvTranspose2d(3,64,kernel_size=4,stride=1,padding=0,bias=False)
2.步长为1没有填充: o = i + (k-1)
      self.g1 = nn.ConvTranspose2d(3,64,kernel_size=4,stride=1,padding=0,bias=False)
3.全填充:o = i - (k -1)
经过试验:
o = (i-1)*s -2p + k

相关文章

网友评论

      本文标题:torch ConvTranspose2d

      本文链接:https://www.haomeiwen.com/subject/ujgpectx.html