轻松学Pytorch – 构建生成对抗网络

OpenCV学堂

发布于 2020-09-01 15:58:05

7910

发布于 2020-09-01 15:58:05

文章被收录于专栏：贾志刚-OpenCV学堂

引言

又好久没有继续写了，这个是我写的第21篇文章，我还在继续坚持写下去，虽然经常各种拖延症，但是我还记得，一直没有敢忘记！今天给大家分享一下Pytorch生成对抗网络代码实现。

什么是生成对抗网络

Ian J. Goodfellow在2014年提出生成对抗网络，从此打开了深度学习中另外一个重要分支，让生成对抗网络（GAN）成为与卷积神经网络（CNN）、循环神经网络（RNN/LSTM）可以并驾齐驱的分支领域。今天GAN仍然是计算机视觉领域研究热点之一，每年还有大量相关的论文产生，GAN已经被用在视觉任务的很多方面，主要包括：

图像合成与数据增广
图像翻译与变换
缺陷检测
图像去噪与重建
图像分割

但是GAN最基本的核心思想还是2014年Ian J. Goodfellow在论文中提到的两个基本的模型分别是:生成器与判别器

生成器（G）：

根据输入噪声Z生成输出样本G(z)
目标：通过生成样本与目标样本分布一致，成功欺骗鉴别器

判别器（D）:

根据输入样本数据来分辨真实样本概率
从数据中学习样本数据的差异性

从a到d，可以看到输入噪声的生成分布越来越接近真实分布X，最终达到一种平衡状态，这种稳定的平衡状态叫纳什均衡，还有一部电影跟这个有关系叫《美丽心灵》。

GAN代码实现

下面的代码实现了基于Mnist数据集实现判别器与生成器，最终通过生成器可以自动生成手写数字识别的图像，输入的z=100是随机噪声，输出的是784个数据表示28x28大小的手写数字样本，损失主要来自两个部分，生成器生成损失，判别器分别判别真实与虚构样本概率，基于反向传播训练两个网络，设置epoch=100，得到最终的生成器生成结果如下：

生成器与判别器代码实现如下

HARR特征级联分类器人脸检测来自VJ的2004论文中提出，其主要思想可以通过下面一张图像解释：

transform = tv.transforms.Compose([tv.transforms.ToTensor(),
                                   tv.transforms.Normalize((0.5,), (0.5,))])
train_ts = tv.datasets.MNIST(root='./data', train=True, download=True, transform=transform)
test_ts = tv.datasets.MNIST(root='./data', train=False, download=True, transform=transform)
train_dl = DataLoader(train_ts, batch_size=128, shuffle=True, drop_last=False)
test_dl = DataLoader(test_ts, batch_size=128, shuffle=True, drop_last=False)


class Generator(t.nn.Module):
    def __init__(self, g_input_dim, g_output_dim):
        super(Generator, self).__init__()
        self.fc1 = t.nn.Linear(g_input_dim, 256)
        self.fc2 = t.nn.Linear(self.fc1.out_features, self.fc1.out_features * 2)
        self.fc3 = t.nn.Linear(self.fc2.out_features, self.fc2.out_features * 2)
        self.fc4 = t.nn.Linear(self.fc3.out_features, g_output_dim)

    # forward method
    def forward(self, x):
        x = F.leaky_relu(self.fc1(x), 0.2)
        x = F.leaky_relu(self.fc2(x), 0.2)
        x = F.leaky_relu(self.fc3(x), 0.2)
        return t.tanh(self.fc4(x))


class Discriminator(t.nn.Module):
    def __init__(self, d_input_dim):
        super(Discriminator, self).__init__()
        self.fc1 = t.nn.Linear(d_input_dim, 1024)
        self.fc2 = t.nn.Linear(self.fc1.out_features, self.fc1.out_features // 2)
        self.fc3 = t.nn.Linear(self.fc2.out_features, self.fc2.out_features // 2)
        self.fc4 = t.nn.Linear(self.fc3.out_features, 1)

    # forward method
    def forward(self, x):
        x = F.leaky_relu(self.fc1(x), 0.2)
        x = F.dropout(x, 0.3)
        x = F.leaky_relu(self.fc2(x), 0.2)
        x = F.dropout(x, 0.3)
        x = F.leaky_relu(self.fc3(x), 0.2)
        x = F.dropout(x, 0.3)
        return t.sigmoid(self.fc4(x))

损失与训练代码如下

分别定义生成网络训练与鉴别网络的训练方法，然后开始训练即可，代码实现如下：

# 生成者与判别者
bs = 128
z_dim = 100
mnist_dim = 784
# loss
criterion = t.nn.BCELoss()

# optimizer
device = "cuda"
gnet = Generator(g_input_dim = z_dim, g_output_dim = mnist_dim).to(device)
dnet = Discriminator(mnist_dim).to(device)
lr = 0.0002
G_optimizer = t.optim.Adam(gnet.parameters(), lr=lr)
D_optimizer = t.optim.Adam(dnet.parameters(), lr=lr)


def D_train(x):
    # =======================Train the discriminator=======================#
    dnet.zero_grad()

    # train discriminator on real
    x_real, y_real = x.view(-1, mnist_dim), t.ones(bs, 1)
    x_real, y_real = Variable(x_real.to(device)), Variable(y_real.to(device))

    D_output = dnet(x_real)
    D_real_loss = criterion(D_output, y_real)

    # train discriminator on facke
    z = Variable(t.randn(bs, z_dim).to(device))
    x_fake, y_fake = gnet(z), Variable(t.zeros(bs, 1).to(device))

    D_output = dnet(x_fake)
    D_fake_loss = criterion(D_output, y_fake)

    # gradient backprop & optimize ONLY D's parameters
    D_loss = D_real_loss + D_fake_loss
    D_loss.backward()
    D_optimizer.step()

    return D_loss.data.item()


def G_train(x):
    # =======================Train the generator=======================#
    gnet.zero_grad()

    z = Variable(t.randn(bs, z_dim).to(device))
    y = Variable(t.ones(bs, 1).to(device))

    G_output = gnet(z)
    D_output = dnet(G_output)
    G_loss = criterion(D_output, y)

    # gradient backprop & optimize ONLY G's parameters
    G_loss.backward()
    G_optimizer.step()

    return G_loss.data.item()


n_epoch = 100
for epoch in range(1, n_epoch+1):
    D_losses, G_losses = [], []
    for batch_idx, (x, _) in enumerate(train_dl):
        bs_, _,_,_ = x.size()
        bs = bs_
        D_losses.append(D_train(x))
        G_losses.append(G_train(x))

    print('[%d/%d]: loss_d: %.3f, loss_g: %.3f' % (
            (epoch), n_epoch, t.mean(t.FloatTensor(D_losses)), t.mean(t.FloatTensor(G_losses))))

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2020-08-27，如有侵权请联系 cloudcommunity@tencent.com 删除

机器学习