我已經從 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()
uj5u.com熱心網友回復:
嘗試使用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()
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/443837.html
標籤:Python 张量流 matplotlib 喀拉斯 生成对抗网络
上一篇:Keras序列模型中的資料增強層
