简介
前面提到的许多生成模型都有保存了生成器的权重,本章主要介绍如何使用训练好的权重文件通过生成器生成图像。
但是如何使用权重生成图像呢?
一、参数配置
ima_size 为图像尺寸,这个需要跟你模型训练的时候resize的时候一样。
latent_dim为噪声维度,一般的设置都是100。
n_classes 为类别数,部分模型有label的输入,所有需要有这个设置。
channels为通道数,这个懂得都懂。
weights_dir为权重文件夹的路径,下面保存的是你的权重文件。
output_dir为权重文件夹生成的图像保存的位置。
img_size = 224
#img_size = 32
latent_dim = 100
n_classes = 10
channels = 3
weights_dir = "weights_NEU05" # 权重文件夹
output_dir = "generated_images" # 生成图像文件夹
二、模型结构
作为GAN模型,必须有生成器的架构,我们的权重文件保存的是生成器的参数,因此在训练好之后得到了生成器的参数文件,通过这个文件我们来生成图像,因此模型架构就是生成器的架构,不需要鉴别器。
定