if hasattr(self.dataset, 'next_generator_sample_test'):
batch = self.dataset.next_generator_sample_test()
else:
batch = self.dataset.next_generator_sample()
print(len(batch))
print(batch[1:].shape)
image = self.dataset.display(self.generator.predict_on_batch(batch), batch)
title = "epoch_{}.png".format(str(self.current_epoch).zfill(3))
if not os.path.exists(self.output_dir):
os.makedirs(self.output_dir)
plt.imsave(os.path.join(self.output_dir, title), image, cmap=plt.cm.gray)`
It is not working with tf 1.12
the error is thrown at
` def save_generated_images(self):