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