12、Pix2Pix与CycleGAN图像转换的实现与应用

Pix2Pix与CycleGAN图像转换的实现与应用

1. Pix2Pix图像到图像转换

1.1 模型构建

在Pix2Pix图像转换中，首先需要创建输入。我们创建两个输入，其形状与输入图像一致，这里图像具有相同的大小、形状和通道数：

self.orig_A = Input(shape=(self.W, self.H, self.C))
self.orig_B = Input(shape=(self.W, self.H, self.C))

接着是条件训练部分，使用 orig_B 作为生成器的输入：

self.fake_A = self.generator.Generator(self.orig_B)

然后构建判别器对象，并将判别器网络设置为不可训练：

self.discriminator = Discriminator(height=self.H, width=self.W, channels=self.C)
self.discriminator.trainable = False

将判别器的输入指向之前创建的 fake_A 和 orig_B 网络：

self.valid =