Gerador convolucional
Defina um gerador convolucional seguindo as diretrizes do DCGAN discutidas no último vídeo.
torch.nn foi pré-importado como nn para facilitar a sua vida. Além disso, tem uma função personalizada dc_gen_block() que retorna um bloco de uma convolução transposta, norma de lote e ativação ReLU. Essa função é tipo um componente básico pra construir o gerador convolucional. Você pode se familiarizar com a definição de “ dc_gen_block() ” abaixo.
def dc_gen_block(in_dim, out_dim, kernel_size, stride):
return nn.Sequential(
nn.ConvTranspose2d(in_dim, out_dim, kernel_size, stride=stride),
nn.BatchNorm2d(out_dim),
nn.ReLU()
)
Este exercício faz parte do curso
Aprendizado profundo para imagens com PyTorch
Instruções do exercício
- Adicione o último bloco gerador, mapeando o tamanho dos mapas de recursos para um
256. - Adiciona uma convolução transposta com o tamanho de saída de
3. - Adicione a ativação tanh.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
class DCGenerator(nn.Module):
def __init__(self, in_dim, kernel_size=4, stride=2):
super(DCGenerator, self).__init__()
self.in_dim = in_dim
self.gen = nn.Sequential(
dc_gen_block(in_dim, 1024, kernel_size, stride),
dc_gen_block(1024, 512, kernel_size, stride),
# Add last generator block
____,
# Add transposed convolution
____(____, ____, kernel_size, stride=stride),
# Add tanh activation
____
)
def forward(self, x):
x = x.view(len(x), self.in_dim, 1, 1)
return self.gen(x)