如何单图像预测

Question

我有一个使用随机图像进行预测的代码。代码在评论部分

**

那是一个随机的图像预测，如何预测单个图像？提前感谢

Answer 1

您需要将图像转换为TF-Dataset或Numpy，然后您就可以按照下面给出的方式馈送到model.prpredict。

im = Image.open(data_path)
im = im.resize((128,128))
im_array = np.array(im)
im_array = np.expand_dims(im_array, axis = 0)
pred = model.predict(im_array)
print("the probable class according to the model is {} with confidence :.2f}".format(class_names[np.argmax(pred)], np.max(pred)*100))