将TensorflowGANS从灰度转换为RGB

Question

我正在编写基于张量流的GAN脚本。它用于灰度图像，但是我无法使生成器适用于RGB在我的代码中由昏暗= 3我可以想象我也必须同时更新添加第三维的其他层，但是如何在密集层进行呢？

model = tf.keras.Sequential()
model.add(layers.Dense( (IMG_SIZE//4) * (IMG_SIZE//4) *256, use_bias=False, input_shape=(100,)))
model.add(layers.BatchNormalization())
model.add(layers.LeakyReLU())

model.add(layers.Reshape((IMG_SIZE//4, IMG_SIZE//4, 256)))
assert model.output_shape == (None, IMG_SIZE//4, IMG_SIZE//4, 256) # Note: None is the batch size

model.add(layers.Conv2DTranspose(128, (5, 5), strides=(1, 1), padding='same', use_bias=False))
assert model.output_shape == (None, IMG_SIZE//4, IMG_SIZE//4, 128)
model.add(layers.BatchNormalization())
model.add(layers.LeakyReLU())

model.add(layers.Conv2DTranspose(64, (5, 5), strides=(2, 2), padding='same', use_bias=False))
assert model.output_shape == (None, IMG_SIZE//2, IMG_SIZE//2, 64)
model.add(layers.BatchNormalization())
model.add(layers.LeakyReLU())

model.add(layers.Conv2DTranspose(1, (5, 5), strides=(2, 2), padding='same', use_bias=False, activation='tanh'))
assert model.output_shape == (None, IMG_SIZE, IMG_SIZE, dim)

return model

Answer 1

这太夸张了，我正在发布另一个问题，因为我不明白如何删除此问题