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使用scrapy采集时伪装成HTTP/1.1的方法
Apr 08 Python
构建Python包的五个简单准则简介
Jun 15 Python
Python 类与元类的深度挖掘 II【经验】
May 06 Python
深入理解python try异常处理机制
Jun 01 Python
python读取图片任意范围区域
Jan 23 Python
详解python selenium 爬取网易云音乐歌单名
Mar 28 Python
Django ORM 聚合查询和分组查询实现详解
Aug 09 Python
python实现截取屏幕保存文件,删除N天前截图的例子
Aug 27 Python
关于Numpy数据类型对象(dtype)使用详解
Nov 27 Python
Python实现钉钉订阅消息功能
Jan 14 Python
利用Python脚本批量生成SQL语句
Mar 04 Python
详解在Python中使用Torchmoji将文本转换为表情符号
Jul 27 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
一个阿拉伯数字转中文数字的函数
2006/10/09 PHP
上传文件先创建目录 再上传到目录里面去
2010/12/29 PHP
三种php连接access数据库方法
2013/11/11 PHP
php生成静态页面的简单示例
2014/04/17 PHP
PHP YII框架开发小技巧之模型(models)中rules自定义验证规则
2015/11/16 PHP
浅谈PHP链表数据结构(单链表)
2016/06/08 PHP
定位地理位置PHP判断员工打卡签到经纬度是否在打卡之内
2019/05/23 PHP
laravel admin实现分类树/模型树的示例代码
2020/06/10 PHP
B/S开发中常用javaScript技术与代码
2007/03/09 Javascript
分享精心挑选的23款美轮美奂的jQuery 图片特效插件
2012/08/14 Javascript
Js制作简单弹出层DIV在页面居中 中间显示遮罩的具体方法
2013/08/08 Javascript
JavaScript动态添加列的方法
2015/03/25 Javascript
AngularJS基础 ng-mouseleave 指令详解
2016/08/02 Javascript
jquery获取input type=text中的值的各种方式(总结)
2016/12/02 Javascript
使用JQ完成表格隔行换色的简单实例
2017/08/25 Javascript
vue2中使用less简易教程
2018/03/27 Javascript
Vue中的Props(不可变状态)
2018/09/29 Javascript
JS实现马赛克图片效果完整示例
2019/04/13 Javascript
Vue+Node实现商品列表的分页、排序、筛选,添加购物车功能详解
2019/12/07 Javascript
微信小程序关键字变色实现代码实例
2019/12/13 Javascript
vue实现移动端项目多行文本溢出省略
2020/07/29 Javascript
jquery自定义组件实例详解
2020/12/31 jQuery
详解Python核心编程中的浅拷贝与深拷贝
2018/01/07 Python
python安装pil库方法及代码
2019/06/25 Python
python 使用装饰器并记录log的示例代码
2019/07/12 Python
Pytorch之contiguous的用法
2019/12/31 Python
如何用Python 加密文件
2020/09/10 Python
python 检测图片是否有马赛克
2020/12/01 Python
python上下文管理器异常问题解决方法
2021/02/07 Python
澳大利亚Rockwear官网:女子瑜伽、健身和运动服
2021/01/26 全球购物
UNOde50美国官网:西班牙珠宝品牌
2020/08/15 全球购物
四年的个人工作自我评价
2013/12/10 职场文书
初三家长会邀请函
2014/01/18 职场文书
我是一名护士演讲稿
2014/08/28 职场文书
导游词之澳门妈祖庙
2019/12/19 职场文书
详解Java实践之建造者模式
2021/06/18 Java/Android