网站分享平台推广电话

简介

前面提到的许多生成模型都有保存了生成器的权重，本章主要介绍如何使用训练好的权重文件通过生成器生成图像。

但是如何使用权重生成图像呢？

一、参数配置

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模型，必须有生成器的架构，我们的权重文件保存的是生成器的参数，因此在训练好之后得到了生成器的参数文件，通过这个文件我们来生成图像，因此模型架构就是生成器的架构，不需要鉴别器。

定