使用Keras加载含有自定义层或函数的模型操作

站长资源 2024/11/24 佚名

2 0 1

当我们导入的模型含有自定义层或者自定义函数时，需要使用custom_objects来指定目标层或目标函数。

例如：

我的一个模型含有自定义层“SincConv1D”，需要使用下面的代码导入：

from keras.models import load_model
model = load_model('model.h5', custom_objects={'SincConv1D': SincConv1D})

如果不加custom_objects指定目标层Layer，则会出现以下报错：

ValueError: Unknown layer: SincConv1D

同样的，当我的模型含有自定义函数“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函数中添加custom_objects参数，来声明自定义的层
（用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加载含有自定义层或函数的模型操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。

Keras,加载,自定义层,函数模型

广告合作：本站广告合作请联系QQ：858582 申请时备注：广告合作（否则不回）
免责声明：本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除！

评论“使用Keras加载含有自定义层或函数的模型操作”

暂无评论...

《魔兽世界》大逃杀！60人新游玩模式《强袭风暴》3月21日上线

暴雪近日发布了《魔兽世界》10.2.6 更新内容，新游玩模式《强袭风暴》即将于3月21 日在亚服上线，届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。

艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕，并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时，他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中，玩家将会进入一个全新的海盗主题大逃杀式限时活动，其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场，作为一个独立于主游戏之外的活动，玩家可以用大逃杀的风格来体验《魔兽世界》，不分职业、不分装备（除了你在赛局中捡到的），光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式，玩家在加入海盗主题的预赛大厅区域前，可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹，《巨龙崛起》和《魔兽世界：巫妖王之怒经典版》的玩家都可以获得奖励。

更新日志

2024年11月24日

使用Keras加载含有自定义层或函数的模型操作

keras 解决加载lstm+crf模型出错的问题

Tensorflow中k.gradients()和tf.stop_gradient()用法说明

评论“使用Keras加载含有自定义层或函数的模型操作”

《魔兽世界》大逃杀！60人新游玩模式《强袭风暴》3月21日上线

更新日志

友情链接