解決Tensorflow2.0 tf.keras.Model.load_weights() 報(bào)錯(cuò)處理問題
錯(cuò)誤描述：
1、保存模型：model.save_weights('./model.h5')
2、腳本重啟
3、加載模型：model.load_weights('./model.h5')
4、模型報(bào)錯(cuò)：ValueError: You are trying to load a weight file containing 12 layers into a model with 0 layers.
問題分析：
模型創(chuàng)建后還沒有編譯，一般是在模型加載前調(diào)用model.build(input_shape), 但我通過Dataset將輸入已經(jīng)變?yōu)閐ict格式了，暫時(shí)沒找這樣輸入怎么匹配input_shape參數(shù)
解決方法：
model.fit(train_dataset, epochs=0)
將epochs設(shè)為0，這樣模型在編譯的同時(shí)不會訓(xùn)練數(shù)據(jù)，減少耗費(fèi)的時(shí)間，之后就可以正常加載保存的參數(shù)了
補(bǔ)充知識：調(diào)用Kears中kears.model.load_model方法遇到的問題和解決方法
之前一直使用tf和pytorch，就算是tf也是tf.estimator用得比較多，很少使用keras，最近嘗試使用kears快速訓(xùn)練和部署一些分類任務(wù),在使用load_model的時(shí)候遇到一些問題
問題1：
SystemError: unknown opcode
原因是因?yàn)槟Ｐ投x用到了lambda
gap = Lambda(lambda x: x[0]/x[1], name = 'RescaleGAP')([gap_features, gap_mask])
我在python3.5的環(huán)境訓(xùn)練的模型，python3.6的環(huán)境load模型。兩個(gè)環(huán)境的lambda有差異，這個(gè)問題。
問題2：
ValueError: Unknown metric function:****
我的錯(cuò)誤是
ValueError: Unknown metric function:top_2_accuracy
因?yàn)樵跇?gòu)建模型時(shí)，使用了自己定義的top_2_accuracy方法，所以在load_model時(shí)需要將top_2_accuracy做為參數(shù)傳進(jìn)去
from keras.models import load_model
from keras.metrics import top_k_categorical_accuracy
def top_2_accuracy(in_gt, in_pred):
return top_k_categorical_accuracy(in_gt, in_pred, k=2)
model = load_model("model.h5",custom_objects={'top_2_accuracy': top_2_accuracy})
以上這篇解決Tensorflow2.0 tf.keras.Model.load_weights() 報(bào)錯(cuò)處理問題就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
堆排序是一種強(qiáng)大的算法,用于在 Python 中對數(shù)組和列表進(jìn)行排序, 它很受歡迎,因?yàn)樗浅？?并且不像合并排序和快速排序那樣占用任何額外空間,本篇文章將介紹堆排序算法在 Python 中的實(shí)現(xiàn),感興趣的朋友跟隨小編一起看看吧2023-08-08
Python編寫一個(gè)驗(yàn)證碼圖片數(shù)據(jù)標(biāo)注GUI程序附源碼
這篇文章主要介紹了Python編寫一個(gè)驗(yàn)證碼圖片數(shù)據(jù)標(biāo)注GUI程序,本文給大家附上小編精心整理的源碼，需要的朋友可以參考下2019-12-12
這篇文章主要介紹了使用Python計(jì)算兩個(gè)不同列表的相似度,Python數(shù)據(jù)開發(fā)工作中的需求兩個(gè)不同列表給出相似度，本文實(shí)現(xiàn)判斷數(shù)字類型相似度和字符串類型相似度，非常實(shí)用,需要的朋友可以參考下2023-07-07
python讀取.mat文件的數(shù)據(jù)及實(shí)例代碼
這篇文章主要介紹了python讀取.mat文件的數(shù)據(jù)的方法，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值 ,需要的朋友可以參考下2019-07-07
使用Python正則表達(dá)式操作文本數(shù)據(jù)的方法
這篇文章主要介紹了使用Python正則表達(dá)式操作文本數(shù)據(jù)的方法，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2019-05-05
這篇文章主要介紹了Pycharm制作搞怪彈窗(聲音強(qiáng)制最大，屏幕亮度強(qiáng)制最亮，按鈕躲避，彈窗炸彈）,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-02-02
python數(shù)據(jù)結(jié)構(gòu)輸入輸出及控制和異常
這篇文章主要介紹了python數(shù)據(jù)結(jié)構(gòu)輸入輸出及控制和異常,上一章節(jié)中我們介紹了python的基礎(chǔ)數(shù)據(jù)類型和集合數(shù)據(jù)類型，這章節(jié)給大家介紹一下python的輸入輸出、控制和異常，對數(shù)據(jù)類型感興趣的同學(xué)可以查看一下文章<BR>2021-12-12
python中k-means和k-means++原理及實(shí)現(xiàn)
本文主要介紹了python中k-means和k-means++原理及實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-05-05
在Python中利用Bokeh創(chuàng)建動態(tài)數(shù)據(jù)可視化
在本文中,我們探討了如何利用 Bokeh 庫在 Python 中創(chuàng)建動態(tài)數(shù)據(jù)可視化,我們介紹了 Bokeh 的基本概念和優(yōu)勢,以及如何安裝 Bokeh 庫,對在Python中利用Bokeh創(chuàng)建動態(tài)數(shù)據(jù)可視化相關(guān)知識感興趣的朋友跟隨小編一起看看吧2024-05-05
Python使用captcha庫制作帶參數(shù)輸入驗(yàn)證碼案例
這篇文章主要介紹了Python使用captcha庫制作驗(yàn)證碼，帶參數(shù)輸入,本文通過實(shí)例案例解析給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-05-05