GAN动漫人物头像生成

GAN动漫人物头像生成

1.简介

搭建了一个简单的DCGAN网络生成动漫人物的头像，其中动漫人物头像数据集取自kaggle，网址如下
link

2.网络结构

1. 数据集
2. 生成器
3. 判别器

2.1数据集

数据大小为64x64x3，样例如下

2.2生成器

由于生成器的原始输入是n维噪声，若想生成与数据集大小相同的图片，则需要进行上采样，这里我们用到的方法是转置卷积，通过pytorch中的ConvTransposed2d来实现。
生成器代码如下：

class Generator(nn.Module):

    def __init__(self, noise_dim=100):
        super(Generator, self).__init__()
        self.net = nn.Sequential(
            # out_shape = (1-1)*1-2*0+4 = 4*4
            nn.ConvTranspose2d(noise_dim, 256, kernel_size=4),
            nn.BatchNorm2d(256),
            nn.ReLU(),
            # out_shape = (4-1)*2-2*1+4 = 8*8
            nn.ConvTranspose2d(256, 128, kernel_size=4, stride=2, padding=1),
            nn.BatchNorm2d(128),
            nn.ReLU(),
            # out_shape = (8-1)*2-2*1+4 = 16*16
            nn.ConvTranspose2d(128, 64, kernel_size=4, stride=2, padding=1),
            nn.BatchNorm2d(64),
            nn.ReLU(),
            # out_shape = (16-1)*2-2*1+4 = 32*32
            nn.ConvTranspose2d(64, 32, kernel_size=4, stride=2, padding=1),
            nn.BatchNorm2d(32),
            nn.ReLU(),
            # out_shape = (32-1)*2-2*1+4 = 64*64
            nn.ConvTranspose2d(