如何使用.pkl文件预测图像

时间:2019-12-12 12:04:40

标签: python scikit-learn deep-learning computer-vision classification

我已经训练了图像分类模型并将其保存在.pkl文件中。
现在我不知道如何使用该泡菜文件进行单个或多个图像预测。 我这样做是为了表格数据预测,而不是图像

from sklearn.externals import joblib

classifer = joblib.load("model.pkl")

下一步该用于表格数据预测的方法不适用于图像。

1 个答案:

答案 0 :(得分:0)

应该相同,您可以使用所使用的任何图像模块读取图像,例如opencv:

import cv2 
img = cv2.imread('path/to/img')
classifier.predict(img)

如果这不起作用,则可以粘贴错误,以便更轻松地为您提供答案