【657】深度学习模型预测单张图片

　　直接用 predict 函数，batch_size 设置为 1

　　代码如下：

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24

# 一个图片预测 from PIL import Image import numpy as np from tensorflow.keras.preprocessing.image import load_img   # 图片转为矩阵输入数据 def img2x(img_path):     img = Image.open(img_path)     x = np.array(img) / 255     x = np.expand_dims(x, axis=0)     return x   img_name = input("Input image name: ") load_img("../models_Res-U-Net/alex_google_map/" + img_name + ".png").show() img_path = "../models_Res-U-Net/alex_google_map/" + img_name + ".png"   x = img2x(img_path) x = x[:,:,:,:3] # 直接通过模型预测，得到输出数据 y = model.predict(x, batch_size=1)[0] # 对于概率数据进行判断得到二值图 mask = (y > 0.5).astype('uint8') img = keras.preprocessing.image.array_to_img(mask) img.show()

 

　　效果如下：