当前位置:主页
> 资源下载 > 50 > 预训练的GAN + VAE +分类器模型在pytorch中,适用于MNIST和CIFAR数据集,项目名为gan-vae-pretrained-pytorch
-
预训练的GAN + VAE +分类器模型在pytorch中,适用于MNIST和CIFAR数据集,项目名为gan-vae-pretrained-pytorch
资源介绍
MNIST / CIFAR10的预训练GAN,VAE +分类器
在pytorch中使用GAN / VAE建模的简单起点。
包括模型类定义+培训脚本
包括笔记本电脑,其中显示了如何加载预训练的网络/如何使用它们
用pytorch 1.0+测试
生成与数据集图像大小相同的图像
mnist
使用基于的体系结构生成MNIST数据集(28x28)大小的图像。 训练了100个纪元。 重量。
数据样本
dcgan样本
样品
为了与不那么复杂的体系结构进行比较,我还在文件夹中包含了一个预训练的非卷积GAN,它基于代码(训练了300个纪元)。
我还基于在文件夹中包含了经过预训练的LeNet分类器,该分类器可实现99%的测试准确性。
cifar10
cifar10 gan来自pytorch示例回购,并实现了DCGAN论文。 它只需要很小的改动就可以生成大小为cifar10数据集(32x32x3)的