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

 更新时间:2020年06月11日 09:28:17   作者:动力橙CHT  
这篇文章主要介绍了keras load model时出现Missing Layer错误的解决方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

问题描述:训练结束后,保存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使用循环实现批量创建文件夹示例

    这篇文章主要介绍了python使用循环实现批量创建文件夹示例,需要的朋友可以参考下
    2014-03-03
  • Django 源码WSGI剖析过程详解

    Django 源码WSGI剖析过程详解

    这篇文章主要介绍了Django 源码WSGI剖析过程详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • Django数据库迁移报错InconsistentMigrationHistory

    Django数据库迁移报错InconsistentMigrationHistory

    最近在使用Django,学习了一下Django数据库迁移,在执行迁移命令时,突然报错,本文就总结了一下原因,感兴趣的小伙伴们可以参考一下
    2021-05-05
  • 如何用python写个模板引擎

    如何用python写个模板引擎

    这篇文章主要介绍了如何用python写个模板引擎,帮助大家更好的理解和使用python,感兴趣的朋友可以了解下
    2021-01-01
  • python实现类的静态变量用法实例

    python实现类的静态变量用法实例

    这篇文章主要介绍了python实现类的静态变量用法,实例分析了Python中基于数组实现静态队列的相关使用技巧,需要的朋友可以参考下
    2015-05-05
  • Python统计节假日剩余天数的脚本

    Python统计节假日剩余天数的脚本

    过完春节,盼着下一个节日,那么如何判断距离节假日还有多少天呢?今天小编给大家介绍使用python脚本来解决这个问题,对Python统计节假日倒计时脚本感兴趣的朋友一起看看吧
    2022-02-02
  • pandas.DataFrame写入数据库的实现方式

    pandas.DataFrame写入数据库的实现方式

    这篇文章主要介绍了pandas.DataFrame写入数据库的实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-08-08
  • python爬虫实现POST request payload形式的请求

    python爬虫实现POST request payload形式的请求

    这篇文章主要介绍了python爬虫实现POST request payload形式的请求,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-04-04
  • python 使用cx-freeze打包程序的实现

    python 使用cx-freeze打包程序的实现

    这篇文章主要介绍了python 使用cx-freeze打包程序的实现,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-03-03
  • Python数据结构之顺序表的实现代码示例

    Python数据结构之顺序表的实现代码示例

    这篇文章主要介绍了Python数据结构之顺序表的实现代码示例,简单介绍了顺序表的相关内容,然后分享了其代码示例,具有一定参考价值,需要的朋友可以了解下。
    2017-11-11

最新评论