基于PyTorch的生成对抗网络实战(7)——利用Pytorch搭建SGAN(Semi-Supervised GAN)生成手写数字并分类

〇、Semi-Supervised GAN介绍

通过强制判别器网络输出分类标签,我们将生成对抗网络(GAN)扩展到半监督学习(Semi-Supervised)。我们在输入属于N类之一的数据集上训练生成器G和判别器D。在训练时,使用D来预测输入属于N+1类中的哪一个,其中一个额外的类被添加到G的输出中。我们表明,这种方法可以用来创建一个更有效的数据分类器,它允许生成比常规GAN更高质量的样本。
论文可参考《Semi-Supervised Learning with Generative Adversarial Networks》。

算法如下图所示:
在这里插入图片描述

一、生成器

生成器的目标是输入一个固定维度的随机噪声,生成手写数字图像,因此它的输入是一个长度为N=100的一

猜你喜欢

转载自blog.csdn.net/didi_ya/article/details/125446580