一张一张保存GAN生成的图片

乘风 tensorflow 279

原文标题 ：Save GAN generated images one by one

我已经从 Fashion Mnist 数据集中生成了一些图像，但是，我无法想出将每个图像保存为单个文件的函数或方法。我只找到了一种将它们分组保存的方法。有人可以帮助我如何一张一张地保存图像吗？

这就是我目前所拥有的：

def generate_and_save_images(model, 
     epoch,test_input):
     predictions = model(test_input, training=False)

     fig = plt.figure(figsize=(4,4))

    for i in range(predictions.shape[0]):
        plt.subplot(4, 4, i+1)
        plt.imshow(predictions[i, :, :, 0] * 127.5 + 
        127.5, cmap='gray')
        plt.axis('off')

    plt.savefig('image_at_epoch_{:04d}.png'.format(epoch))

    plt.show()

原文链接：https://stackoverflow.com//questions/71452209/save-gan-generated-images-one-by-one

回复

我来回复
  • AloneTogether的头像
    AloneTogether 评论

    尝试使用plt.imsave分别保存每个图像：

    def generate_and_save_images(model, epoch, test_input):
  predictions = model(test_input, training=False)
  fig = plt.figure(figsize=(4, 4))

  for i in range(predictions.shape[0]):
      plt.subplot(4, 4, i+1)
      plt.imshow(predictions[i, :, :, 0] * 127.5 + 127.5, cmap='gray')
      plt.imsave('image_at_epoch_{:04d}-{}.png'.format(epoch, i), predictions[i, :, :, 0] * 127.5 + 127.5, cmap='gray')
      plt.axis('off')

  plt.savefig('image_at_epoch_{:04d}.png'.format(epoch))
  plt.show()
    2年前 0条评论

相关问题