Scikit-learn:使用joblib
库保存模型,文件后缀通常是.pkl
或.joblib
用joblib加载模型
model = joblib.load('app01/static/svm.pkl')
img_res = model.predict(img)
TensorFlow/Keras:模型可以保存为.h5
或.hdf5
文件,这些是HDF5格式的文件,用于存储模型结构和权重。
用models加载模型
model = models.load_model('model/cotton.h5')
print(model.predict(image))