FashionGen 上的 DCGAN

import torch
use_gpu = True if torch.cuda.is_available() else False
model = torch.hub.load('facebookresearch/pytorch_GAN_zoo:hub', 'DCGAN', pretrained=True, useGPU=use_gpu)
模型的输入是一个形状为 (N, 120)
的噪声向量,其中 N
是要生成的图像数量。可以使用 .buildNoiseData
函数来构建它。模型有一个 .test
函数,它接收噪声向量并生成图像。
num_images = 64
noise, _ = model.buildNoiseData(num_images)
with torch.no_grad():
generated_images = model.test(noise)
# let's plot these images using torchvision and matplotlib
import matplotlib.pyplot as plt
import torchvision
plt.imshow(torchvision.utils.make_grid(generated_images).permute(1, 2, 0).cpu().numpy())
# plt.show()
您应该会看到一张与左侧类似的图像。
如果您想从头开始训练自己的 DCGAN 和其他 GAN,请查看 PyTorch GAN Zoo。
模型描述
在计算机视觉中,生成模型是经过训练的神经网络,用于根据给定输入创建图像。在我们的例子中,我们考虑一种特定类型的生成网络:GAN(生成对抗网络),它学习将随机向量映射到逼真的图像生成。
DCGAN 是 Radford 等人于 2015 年在论文 Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks 中设计的模型。它是一种 GAN 架构,对于低分辨率图像生成(高达 64×64)来说既简单又高效。
要求
- 目前仅支持 Python 3
参考资料
