如何將 CNN 預測保存為 float32 數據類型的圖像？

Question

我正在嘗試將 CNN 預測結果 output 為 float32 格式。 當我使用以下代碼時，圖像輸出以 8 位格式在 0-255 之間縮放。 我怎樣才能保持float32？

pred= model.predict(X_test)

-> dtype('float32')

for i, image in enumerate(pred, 1):
    tf.keras.preprocessing.image.save_img(f'output_test/{i}.tif', image,scale=False)

Answer 1

為了將預測保存到 float32 數據類型的圖像，我們可以使用tifffile 。 編輯后的代碼是：

import numpy as np
from tifffile import imsave

pred= model.predict(X_test)

for i, image in enumerate(pred, 1):
    imsave(f'output_test/{i}.tif', image)

Answer 2

嘗試這個：

from keras.preprocessing.image import array_to_img
from PIL import Image

pred = model.predict(X_test)
for i, im in enumerate(pred, 1):
    img = array_to_img(im, scale=False)
    img.save('saved_image_{}.png'.format(i), format="PNG")