因圖像顯示的0-255範圍的,所以對於0,1值得圖像顯示就出現全黑的情況,那麼如何將全黑的圖像轉化到0-255範圍內,從而正常顯示出黑白的二值圖。實現代碼如下:
import tensorflow as tf import numpy as np from glob import glob import cv2 if __name__ == '__main__': # Open existing project on disk. test_images = glob('E:/圖像數據集/output/masks/*') print(test_images) # Imgpath='41253.png' for img in test_images: # print(img) image = tf.io.read_file(img) Img = tf.image.decode_image(image, channels=3) Cmax = np.max(Img[:]) Cmin = np.min(Img[:]) Img = np.uint8((np.double(Img) - Cmin) / (Cmax - Cmin) * 255) fname = img.split('\\')[-1].split('.')[0] + '.png' cv2.imwrite(f'./1024/{fname}', Img)
轉化後