Генерация рентгеновского изображения грудной клетки с помощью GAN

Question

Я новичок в GAN и использую его для создания большего количества рентгеновских снимков грудной клетки, но у меня возникла проблема с сохранением этих сгенерированных изображений как отдельного изображения, а не в виде сетки. Изображения GAN

Так как я могу сохранить все сгенерированные изображения как отдельные изображения Код, который я использую для сохранения его в виде сетки, приведен ниже

def sample_images(epoch):
  rows, cols = 10, 10
  noise = np.random.randn(rows * cols, latent_dim)
  imgs = generator.predict(noise)

  # Rescale images 0 - 1
  imgs = 0.5 * imgs + 0.5

  fig, axs = plt.subplots(rows, cols)
  idx = 0
  for i in range(rows):
    for j in range(cols):
      axs[i,j].imshow(imgs[idx].reshape(H, W), cmap='gray')
      axs[i,j].axis('off')
      idx += 1

  fig.savefig("normal_gan_images/%d.png" % epoch)
  plt.close()

Answer 1

fig в fig, axs = plt.subplots(rows, cols) представляет весь участок. Таким образом, вы сохраняете всю сетку в результате.

Итак, попробуйте это:

def sample_images(epoch):
  rows, cols = 10, 10
  noise = np.random.randn(rows * cols, latent_dim)
  imgs = generator.predict(noise)

  # Rescale images 0 - 1
  imgs = 0.5 * imgs + 0.5

  for i in range(imgs.shape[0]):
      plt.imshow(imgs[i])
      plt.savefig(f"normal_gan_images/{epoch}_{i}.png")
      plt.close()