DCGAN 실습 Generator 질문
미해결
[PyTorch] 쉽고 빠르게 배우는 GAN
안녕하세요. DCGAN 실습부분 Generator 부분의 shape가 궁금해서 질문드립니다. 아래 코드 부분에 nn.ConvTranspose2d가 어떻게 동작해서 state size가 4x4, 8x8, 16x16이 되는 지 궁금합니다. (ngf*8) x 4 x 4 (ngf*4) x 8 x 8 (ngf*2) x 16 x 16 class Generator ( nn . Module ) : def __init__ ( self ) : super ( Generator , self ) . __init__ () self .main = nn.Sequential ( # input is Z, going into a convolution nn.ConvTranspose2d ( nz , ngf * 8 , 4 , 1 , 0 , bias= False ), nn.BatchNorm2d ( ngf * 8 ), nn.ReLU ( True ), # state size. (ngf*8) x 4 x 4 nn.ConvTranspose2d ( ngf * 8 , ngf * 4 , 4 , 2 , 1 , bias= False ), nn.BatchNorm2d ( ngf * 4 ), nn.ReLU ( True ), # state size. (ngf*4) x 8 x 8 nn.ConvTranspose2d ( ngf * 4 , ngf * 2 , 4 , 2 , 1 , bias= False ), nn.BatchNorm2d ( ngf * 2 ), nn.ReLU ( True ), # state size. (ngf*2) x 16 x 16 nn.ConvTranspose2d ( ngf * 2 , ngf , 4 , 2 , 1 , bias= False ), nn.BatchNorm2d ( ngf ), nn.ReLU ( True ), # state size. (ngf) x 32 x 32 nn.ConvTranspose2d ( ngf , nc , 4 , 2 , 1 , bias= False ), nn.Tanh () # state size. (nc) x 64 x 64 )
- dcgan
- pytorch
- 인공신경망
- 딥러닝