Блог

Я запускаю код для отображения скрытых космических изображений, но цвет изображения не отображается как RGB. Если вы знаете, не могли бы вы сказать мне, в чем причина этого результата? Я хотел бы выводить каждое изображение следующим образом

Код проблемы и результат являются

 random_latent_vectors = tf.random.normal(shape=(10, 128))
generator = make_generator(128)

images = generator(random_latent_vectors)
images *= 255

images = images.numpy()
images.shape

plt.figure(figsize=(8,3))
for i in range(images.shape[0]):
  plt.subplot(2, 5, i 1)
  plt.imshow(images[i, :, :, 0].astype("int32"))
  plt.axis('off')
plt.show()
 

generator построен из

 def make_generator(latent_dim):
  model = keras.Sequential([
                            keras.Input(shape=(latent_dim,)),
                            layers.Dense(8 * 8 * 128),
                            layers.Reshape((8, 8, 128)),
                            layers.Conv2DTranspose(128, kernel_size=4, strides=2, padding="same"),
                            layers.LeakyReLU(alpha=0.2),

                            layers.Conv2DTranspose(256, kernel_size=4, strides=2, padding="same"),
                            layers.LeakyReLU(alpha=0.2),
                            
                            layers.Conv2DTranspose(512, kernel_size=4, strides=2, padding="same"),
                            layers.LeakyReLU(alpha=0.2),
                            layers.Conv2D(3, kernel_size=5, padding="same", activation="sigmoid")
                            ])
  return model