試したこと
33→55のアップサンプリングをConvTranspose2dで実現するために、
まずConv2dによる逆の処理(55→33)の理解が必要と考え、下記の通り実行しました。
(こちらのサイト様等参考にしました)
Python
1import torch 2import torch.nn as nn 3 4in_channels = 1 5out_channels = 1 6kernel_size = 2 7stride = 2 8padding = 1 9 10conv = nn.Conv2d(in_channels, out_channels,kernel_size,stride,padding) 11 12x = torch.rand(1, 1, 5, 5) 13print(conv(x).size()) # torch.Size([1, 1, 3, 3]) 14
上記のパラメータの条件で55配列が33配列になったので、これを参考に55配列にConvTranspose2dを実行したところ、44の配列が出力されました。
Python
1import torch 2import numpy as np 3 4#3*3の入力 5Input = torch.tensor(np.array([[[[2.,3.,2.],[1.,2.,4.],[4.,2.,1.]]]])) 6 7#2*2のカーネル 8Kernel = torch.tensor(np.array([[[[2.,1.],[1.,2.]]]])) 9 10Conv = torch.nn.ConvTranspose2d(in_channels=1, 11 out_channels=1, 12 kernel_size=2, 13 stride=2, 14 padding=1,#実際の幅はdilation *(kernel_size - 1) - padding = 0 ? 15 output_padding=0, 16 bias = False) 17 18 19Conv.weight.data = Kernel 20 21Output = Conv(Input) 22print(Output.size())#torch.Size([1, 1, 4, 4]) 23
聞きたいこと
3*3に出力する場合はどのようにパラメータを調整すればよいでしょうか?
理解不足で恐縮ですが、ご教示いただけますと幸いです。
あなたの回答
tips
プレビュー