Keras 使用:加载已经训练好的模型进行预测

版权声明:本博客内容为原创,若要转载,请注明出处!否则禁止转载! https://blog.csdn.net/wardenjohn/article/details/84063284

使用Keras训练好的模型用来直接进行预测,这个时候我们该怎么做呢?【我这里使用的就是一个图片分类网络】

现在让我来说说怎么样使用已经训练好的模型来进行预测判定把

首先,我们已经又有了model模型,这个模型被保存为model.h5文件

然后我们需要在代码里面进行加载

model = load_model("model.h5")

假设我们自己已经写好了一个load_data函数【load_data最好是返回已经通过了把图片转成numpy的data,以及图片对应的label】

然后我们先加载我们的待预测的数据

data, labels = load_data(<the path of the data>)

然后我们就可以通过模型来预测了

predict = model.predict(data)

得到的predict就是预测的结果啦

猜你喜欢

转载自blog.csdn.net/wardenjohn/article/details/84063284