使用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
投稿猜你喜欢
首先呢，我们来看看一般项目路由是怎么划分的。为什么这么划分呢？如果大项目业务非常多，单纯的单页面很难维护，我们只有这样规范化，才能高效率。模- 本文实例讲述了python实现自动更换ip的方法。分享给大家供大家参考。具体实现方法如下：#!/usr/bin/env python#-*-
- 本文主要讲解的是表单，这个其实对于做过网站的人来说，并不陌生，而且可以说是最为常用的提交数据的Form表单。本文主要来讲解一下内容：1.基本
- 在Python中有一些内置的数据类型，比如int, str, list, tuple, dict等。Python的collections模块
- 不管何时何地，只要我们编程时遇到了跟时间有关的问题，都要想到 datetime 和 time 标准库模块，今天我们就用它内部的方法，详解py
- 本文实例讲述了Python基于pillow判断图片完整性的方法。分享给大家供大家参考，具体如下：1、安装第三方库。pip install p
- #coding=utf8__author__ = 'Administrator'# 当函数的参数不确定时，可以使用*args
- 在 JavaScript 中对象和数组是引用类型，指向同一个内存空间，如果 prop 是一个对象或数组，在子组件内部改变它会影响父组件的状态
- 工具安装主要调用win32库实现分辨率获取和读写，需要安装pywin32示例中是从execl列表中读取需要设置的分辨率，需要安装xlrd用到
- 最近在研究网页的切片算法，很可能很多人不知道什么是切片算法，其实这是一种面向搜索引擎的网页分块、切片的原理，目前随着工作的深入，逐渐碰到了各
- 在脚本中可设置需要备份的数据库表清单，并且会将备份文件通过gzip压缩。需要注意的是，这段脚本仅适用数据一致性要求不高的环境。#!/bin/
- LCase：转成小写 UCase：转成大写 下面是ASP中的代码，可以直接演示效果的。 代码如下:<％ dim s
- 一、数据合并与分割1.tf.concat()填入两个tensor, 指定某维度，在指定的维度合并。除了合并的维度之外，其他的维度必须相等。2
- 示例：# -*- coding:utf-8 -*-import jsonstrtest = {"中故宫":"好
- 一、介绍正则表达式是一个特殊的字符序列，计算机科学的一个概念。通常被用来检索、替换那些符合某个模式(规则)的文本。许多程序设计语言都支持利用
- 1. 先介绍几个常用的 MySQL 函数RAND() 随机生成 0~1 之间的小数（0<1）CEI
- JPA之映射mysql text类型问题背景jpa如果直接映射mysql的text/longtext/tinytext类型到String字段
- 在SQL SERVER中，你可能需要获得当前日期和计算一些其他的日期，例如，你的程序可能需要判断一个月的第一天或者最后一天。你们大部分人大概
- 简 介： 根据文档安装指南 介绍了在本机安装PaddlePaddle的步骤。关键词： PaddlePaddle，pip，安装§01 安装Pa
- 0、前言在python2.7及以上的版本，str.format()的方式为格式化提供了非常大的便利。与之前的％型格式化字符串相比，他显得更为
