Keras 加載已經(jīng)訓(xùn)練好的模型進行預(yù)測操作
使用Keras訓(xùn)練好的模型用來直接進行預(yù)測,這個時候我們該怎么做呢?【我這里使用的就是一個圖片分類網(wǎng)絡(luò)】
現(xiàn)在讓我來說說怎么樣使用已經(jīng)訓(xùn)練好的模型來進行預(yù)測判定把
首先,我們已經(jīng)又有了model模型,這個模型被保存為model.h5文件
然后我們需要在代碼里面進行加載
model = load_model("model.h5")
假設(shè)我們自己已經(jīng)寫好了一個load_data函數(shù)【load_data最好是返回已經(jīng)通過了把圖片轉(zhuǎn)成numpy的data,以及圖片對應(yīng)的label】
然后我們先加載我們的待預(yù)測的數(shù)據(jù)
data, labels = load_data(<the path of the data>)
然后我們就可以通過模型來預(yù)測了
predict = model.predict(data)
得到的predict就是預(yù)測的結(jié)果啦~
補充知識:keras利用vgg16模型直接預(yù)測圖片類型時的坑
第一次使用keras中的預(yù)訓(xùn)練模型時,若本地沒有模型對應(yīng)的h5文件,程序會自動去github上下載,但國內(nèi)下載github資源速度太慢,
可以選擇直接去搜索下載,下載后將模型(h5文件)放入C:\Users\lovemoon\.keras\models
同樣,如果是第一個用預(yù)訓(xùn)練模型預(yù)測輸入圖片,解碼結(jié)果時也會下載一個Json文件,同樣可以手動下載后放入C:\Users\lovemoon\.keras\models
以上這篇Keras 加載已經(jīng)訓(xùn)練好的模型進行預(yù)測操作就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python Unittest ddt數(shù)據(jù)驅(qū)動的實現(xiàn)
本文主要介紹了Python Unittest ddt數(shù)據(jù)驅(qū)動的實現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02selenium+python實現(xiàn)基本自動化測試的示例代碼
這篇文章主要介紹了selenium+python實現(xiàn)基本自動化測試的示例代碼,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01Python+OpenCV實現(xiàn)尋找到圓點標(biāo)定板的角點
這篇文章主要為大家詳細(xì)介紹了Python+OpenCV實現(xiàn)找到圓點標(biāo)定板所有點后通過距離找兩個角點,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解一下2022-11-11