使用Keras加载含有自定义层或函数的模型操作
作者：z小白 发布时间：2022-12-25 19:29:29
标签：Keras,加载,自定义层,函数
当我们导入的模型含有自定义层或者自定义函数时，需要使用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})
来源：https://blog.csdn.net/zzc15806/article/details/88222125
0
投稿猜你喜欢
- 近期接手越来越多的东西，发布和运维的工作相当机械，加上频率还蛮高，导致时间浪费还是优点多。修复bug什么的，测试，提交版本库(2分钟)，ss
pip install psycopg2出现错误：Looking in indexes: https://pypi.tuna.tsinghu- 如何在刷新链接之前验证文件是否存在？如何在每次刷新链接之前，验证链接文件是否确实存在？特别是对于数据库中的文件，我们必须在处理它之前确认一下
- Altova 公司的 XMLSPY 是个不可多得的好东西，它几乎可以开发所有的 XML 产品。最近用它来做 Schema
- 导入模块from bs4 import BeautifulSoupsoup = BeautifulSoup(html_doc,"h
- 复制目录: 包含多层子目录方法: 递归, 深度遍历,广度遍历深度遍历&广度遍历:思路:1.获得源目录子级目录,并设置目标目录的子级路
- 高级语言不能直接被机器所理解执行，所以都需要一个翻译的阶段，解释型语言用到的是解释器，编译型语言用到的是编译器。编译型语言通常的执行过程是：
- 目录一、axis简介二、不一样的axis对于axis=0三、总结补充：python中某些函数axis参数的理解在我们使用Python中的Nu
- MNIST数据集比较小，一般入门机器学习都会采用这个数据集来训练下载地址：yann.lecun.com/exdb/mnist/有4个有用的文
- 这篇文章主要介绍了python导入不同目录下的自定义模块过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值
- 之前在使用Pandas处理csv文件时，发现如果文件名为中文，则会报错：OSError: Initializing from file fa
- 支付宝或者微信支付导出的收款二维码，除了二维码部分，还有很大一块背景图案，例如下面就是微信支付的收款二维码：有时候我们仅仅只想要图片中间的方
- 本文实例讲述了Python从序列中移除重复项且保持元素间顺序不变的方法。分享给大家供大家参考，具体如下：问题：从序列中移除重复的元素，但仍然
- 译者：AlphaImageLoader是一个让IE6正常显示PNG32时要用到的一个滤镜，但它在使用中也会产生一系列的问题，本文对使用Alp
- 效果图最近帮朋友写个简单爬虫，顺便整理了下，搞成了一个带GUI界面的小说爬虫工具，用来从笔趣阁爬取小说。开发完成后的界面采集过程界面采集后存
- 前言在Python的世界里，将一个对象以json格式进行序列化或反序列化一直是一个问题。Python标准库里面提供了json序列化的工具，我
- 简介LRU(Least Recently Used)最近最少使用，最近有时间和空间最近的歧义，所以我更喜欢叫它近期最少使用算法。它的核心思想
- matplotlib中的字体文件被封装在font_manager这个子模块中，fontManager.ttflist这个列表涵盖了所有Mat
- 中国，美国，英国3国时间js同步动态显示，对于做企业网站的朋友相信用的到，特别是做英文网站的朋友，加上这一段代码会给你的网站增色不少！本文j
- 这篇文章主要介绍了python调用摄像头的示例代码，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下一、打开摄像头import
