上传者: 42131424
|
上传时间: 2021-10-10 13:40:14
|
文件大小: 88.1MB
|
文件类型: -
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)的