keras load model时出现Missing Layer错误的解决方式


Posted in Python onJune 11, 2020

问题描述:训练结束后,保存model为hdf5和yaml格式的文件

yamlFilename = os.path.join(dir,filename)
yamlModel = model.toyaml()
with open(yamlFilename, "w") as yamlFile:
 yamlFile.write(yamlModel)

随后load model

with open(chkptFilename,'r') as f:
 model_yaml = f.read()
model = KM.model_from_yaml(model_yaml,customs_objects={"dict":dict})
model.load_weights(weightFilename)

但是报错

问题分析:

经过debug分析,原因出在model建立过程中前面lambda层的inbound_node列表中含有后面层,因此从上到下load时,会找不到后面层。重新建立一次model,然后用model.summary() 可以看出其中的原因。

出现这种情况,可能的原因在于,该lambda层在其他py文件中定义,然后import进来,前后多次用到这个lambda层的话,在模型编译过程中,该lambda层可能只编译了一次,前后层共用之,导致后面层结点出现在前面层的inbound_node列表中。

解决办法:

不要在其他py文件中自定义lambda层,直接将其定义在model建立的文件中。或者直接继承Layer层,在其他py文件中重新自定义该层。

补充知识:加载keras模型'tf' is not defined on load_model() - using lambda NameError: name 'tf' is not defined报错

解决方法如下:

import tensorflow as tf
import keras
model = keras.models.load_model('my_model.h5', custom_objects={'tf': tf})

以上这篇keras load model时出现Missing Layer错误的解决方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中.join()和os.path.join()两个函数的用法详解
Jun 11 Python
Python补齐字符串长度的实例
Nov 15 Python
Python正则表达式实现简易计算器功能示例
May 07 Python
Laravel框架表单验证格式化输出的方法
Sep 25 Python
手动安装python3.6的操作过程详解
Jan 13 Python
解决Python命令行下退格,删除,方向键乱码(亲测有效)
Jan 16 Python
Python3之乱码\xe6\x97\xa0\xe6\xb3\x95处理方式
May 11 Python
使用 prometheus python 库编写自定义指标的方法(完整代码)
Jun 29 Python
解决python 虚拟环境删除包无法加载的问题
Jul 13 Python
Python描述数据结构学习之哈夫曼树篇
Sep 07 Python
python 利用Pyinstaller打包Web项目
Oct 23 Python
教你如何使用Python下载B站视频的详细教程
Apr 29 Python
Pyinstaller加密打包应用的示例代码
Jun 11 #Python
解决keras加入lambda层时shape的问题
Jun 11 #Python
python opencv把一张图片嵌入(叠加)到另一张图片上的实现代码
Jun 11 #Python
keras slice layer 层实现方式
Jun 11 #Python
Spring http服务远程调用实现过程解析
Jun 11 #Python
keras Lambda自定义层实现数据的切片方式,Lambda传参数
Jun 11 #Python
python怎么提高计算速度
Jun 11 #Python
You might like
php pki加密技术(openssl)详解
2013/07/01 PHP
在Ubuntu 14.04上部署 PHP 环境及 WordPress
2014/09/02 PHP
php+xml实现在线英文词典之添加词条的方法
2015/01/23 PHP
CentOS下PHP7的编译安装及MySQL的支持和一些常见问题的解决办法
2015/12/17 PHP
Windows 下安装 swoole 图文教程(php)
2017/06/05 PHP
特殊字符、常规符号及其代码对照表
2006/06/26 Javascript
动态加载js、css等文件跨iframe实现
2014/02/24 Javascript
javascript简单实现图片预加载
2014/12/03 Javascript
jQuery循环动画与获取组件尺寸的方法
2015/02/02 Javascript
SuperSlide标签切换、焦点图多种组合插件
2015/03/14 Javascript
jQuery实现动画效果circle实例
2015/08/06 Javascript
jquery实现页面虚拟键盘特效
2015/08/08 Javascript
使用JQuery FancyBox插件实现图片展示特效
2015/11/16 Javascript
AngularJS打开页面隐藏显示表达式用法示例
2016/12/25 Javascript
js调用设备摄像头的方法
2018/07/19 Javascript
ES6的异步终极解决方案分享
2019/07/11 Javascript
详解在IDEA中将Echarts引入web两种方式(使用js文件和maven的依赖导入)
2020/07/11 Javascript
python使用pil生成图片验证码的方法
2015/05/08 Python
关于numpy中np.nonzero()函数用法的详解
2017/02/07 Python
python编程羊车门问题代码示例
2017/10/25 Python
python实现TF-IDF算法解析
2018/01/02 Python
python爬虫自动创建文件夹的功能
2018/08/01 Python
用Python实现筛选文件脚本的方法
2018/10/27 Python
python标记语句块使用方法总结
2019/08/05 Python
pytorch之inception_v3的实现案例
2020/01/06 Python
tensorflow的ckpt及pb模型持久化方式及转化详解
2020/02/12 Python
TensorFlow tf.nn.softmax_cross_entropy_with_logits的用法
2020/04/19 Python
完美解决TensorFlow和Keras大数据量内存溢出的问题
2020/07/03 Python
详解python中的lambda与sorted函数
2020/09/04 Python
通过css3动画和opacity透明度实现呼吸灯效果
2019/08/09 HTML / CSS
css3简单练习实现遨游浏览器logo的绘制
2013/01/30 HTML / CSS
请编写一个 C 函数,该函数在给定的内存区域搜索给定的字符,并返回该字符所在位置索引值
2014/09/15 面试题
优秀毕业生自荐信范文
2014/01/01 职场文书
高中微机老师自我鉴定
2014/02/16 职场文书
平面设计专业求职信
2014/08/09 职场文书
乡镇三严三实学习心得体会
2014/10/13 职场文书