使用Keras加載含有自定義層或函數(shù)的模型操作
當我們導入的模型含有自定義層或者自定義函數(shù)時，需要使用custom_objects來指定目標層或目標函數(shù)。
例如：
我的一個模型含有自定義層“SincConv1D”，需要使用下面的代碼導入：
from keras.models import load_model
model = load_model('model.h5', custom_objects={'SincConv1D': SincConv1D})
如果不加custom_objects指定目標層Layer，則會出現(xiàn)以下報錯：
ValueError: Unknown layer: SincConv1D
同樣的，當我的模型含有自定義函數(shù)“my_loss”，需要使用下面的代碼導入：
from keras.models import load_model
model = load_model('model.h5', custom_objects={'my_loss': my_loss})
補充知識：keras加載模型load_model報錯——ValueError: Unknown layer: CRF
我就廢話不多說了，大家還是直接看代碼吧！
from keras.models import load_model
model = load_model(model_path)
會報錯，需要在load_model函數(shù)中添加custom_objects參數(shù)，來聲明自定義的層
（用keras搭建bilstm-crf，在訓練模型時，使用的是：
from keras_contrib.layers.crf import CRF）
from keras_contrib.layers.crf import CRF, crf_loss, crf_viterbi_accuracy
model = load_model(model_path, custom_objects={"CRF": CRF, 'crf_loss': crf_loss,
'crf_viterbi_accuracy': crf_viterbi_accuracy})
以上這篇使用Keras加載含有自定義層或函數(shù)的模型操作就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了如何修改numpy?array的數(shù)據(jù)類型，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
這篇文章將詳細講解圖像灰度非線性變換。圖像灰度非線性變換主要包括對數(shù)變換、冪次變換、指數(shù)變換、分段函數(shù)變換，通過非線性關(guān)系對圖像進行灰度處理，本文主要講解三種常見類型的灰度非線性變換，感興趣的可以了解一下2022-03-03
Python實現(xiàn)周期性抓取網(wǎng)頁內(nèi)容的方法
這篇文章主要介紹了Python實現(xiàn)周期性抓取網(wǎng)頁內(nèi)容的方法,涉及Python時間函數(shù)及正則匹配的相關(guān)操作技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-11-11
selenium+opencv實現(xiàn)滑塊驗證碼的登陸
很多網(wǎng)站登錄登陸時都要用到滑塊驗證碼，本文主要介紹了selenium+opencv實現(xiàn)滑塊驗證碼的登陸，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-04-04
關(guān)于python 的legend圖例,參數(shù)使用說明
這篇文章主要介紹了關(guān)于python 的legend圖例,參數(shù)使用說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04
解決python字典對值(值為列表)賦值出現(xiàn)重復的問題
今天小編就為大家分享一篇解決python字典對值(值為列表)賦值出現(xiàn)重復的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
Django中在xadmin中集成DjangoUeditor過程詳解
這篇文章主要介紹了Django中在xadmin中集成DjangoUeditor過程詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-07-07
在本篇文章里小編給大家整理的是一篇關(guān)于python析構(gòu)函數(shù)用法及注意事項，有需要的朋友們可以學習參考下。2021-06-06
這篇文章主要介紹了Django和Flask框架相關(guān)對比,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-10-10
最近有粉絲向我咨詢super相關(guān)的問題，說網(wǎng)上搜索到的教程不夠通俗易懂，看了之后還是不太理解。所以在這里基于我自己的理解來講解一下super2021-10-10