1. 程式人生 > >Keras 使用:載入已經訓練好的模型進行預測

Keras 使用:載入已經訓練好的模型進行預測

使用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就是預測的結果啦