解决Tensorflow2.0 tf.keras.Model.load_weights() 报错处理问题


Posted in Python onJune 12, 2020

错误描述:

1、保存模型:model.save_weights('./model.h5')

2、脚本重启

3、加载模型:model.load_weights('./model.h5')

4、模型报错:ValueError: You are trying to load a weight file containing 12 layers into a model with 0 layers.

问题分析:

模型创建后还没有编译,一般是在模型加载前调用model.build(input_shape), 但我通过Dataset将输入已经变为dict格式了,暂时没找这样输入怎么匹配input_shape参数

解决方法:

model.fit(train_dataset, epochs=0)

将epochs设为0,这样模型在编译的同时不会训练数据,减少耗费的时间,之后就可以正常加载保存的参数了

补充知识:调用Kears中kears.model.load_model方法遇到的问题和解决方法

之前一直使用tf和pytorch,就算是tf也是tf.estimator用得比较多,很少使用keras,最近尝试使用kears快速训练和部署一些分类任务,在使用load_model的时候遇到一些问题

问题1:

SystemError: unknown opcode

原因是因为模型定义用到了lambda

gap = Lambda(lambda x: x[0]/x[1], name = 'RescaleGAP')([gap_features, gap_mask])

我在python3.5的环境训练的模型,python3.6的环境load模型。两个环境的lambda有差异,这个问题。

问题2:

ValueError: Unknown metric function:****

我的错误是

ValueError: Unknown metric function:top_2_accuracy

因为在构建模型时,使用了自己定义的top_2_accuracy方法,所以在load_model时需要将top_2_accuracy做为参数传进去

from keras.models import load_model
from keras.metrics import top_k_categorical_accuracy
 
def top_2_accuracy(in_gt, in_pred):
  return top_k_categorical_accuracy(in_gt, in_pred, k=2)
 
model = load_model("model.h5",custom_objects={'top_2_accuracy': top_2_accuracy})

以上这篇解决Tensorflow2.0 tf.keras.Model.load_weights() 报错处理问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python 获取et和excel的版本号
Apr 09 Python
Python笔记(叁)继续学习
Oct 24 Python
Python批量修改文件后缀的方法
Jan 26 Python
在Python中使用zlib模块进行数据压缩的教程
Jun 26 Python
使用Python通过win32 COM打开Excel并添加Sheet的方法
May 02 Python
在python中bool函数的取值方法
Nov 01 Python
python爬取酷狗音乐排行榜
Feb 20 Python
Python matplotlib学习笔记之坐标轴范围
Jun 28 Python
用python解压分析jar包实例
Jan 16 Python
什么是python类属性
Jun 10 Python
细说NumPy数组的四种乘法的使用
Dec 18 Python
python数字图像处理之图像自动阈值分割示例
Jun 28 Python
python + selenium 刷B站播放量的实例代码
Jun 12 #Python
解决Keras自带数据集与预训练model下载太慢问题
Jun 12 #Python
keras导入weights方式
Jun 12 #Python
keras读取h5文件load_weights、load代码操作
Jun 12 #Python
Python matplotlib 绘制双Y轴曲线图的示例代码
Jun 12 #Python
keras的siamese(孪生网络)实现案例
Jun 12 #Python
基于python实现模拟数据结构模型
Jun 12 #Python
You might like
php preg_replace替换实例讲解
2013/11/04 PHP
PHP中的事务使用实例
2015/05/26 PHP
php使用ffmpeg向视频中添加文字字幕的实现方法
2016/05/23 PHP
PHP查看SSL证书信息的方法
2016/09/22 PHP
JavaScript 检测浏览器和操作系统的脚本
2008/12/26 Javascript
jQuery AnythingSlider滑动效果插件
2010/02/07 Javascript
javascript JSON操作入门实例
2010/04/16 Javascript
jQuery实现的立体文字渐变效果
2010/05/17 Javascript
jquery.ajax的url中传递中文乱码问题的解决方法
2014/02/07 Javascript
Javascript实现简单的富文本编辑器附演示
2014/06/16 Javascript
jQuery中:gt选择器用法实例
2014/12/29 Javascript
javascript上下方向键控制表格行选中并高亮显示的方法
2015/02/13 Javascript
浅谈$(document)和$(window)的区别
2015/07/15 Javascript
浅谈Cookie的生命周期问题
2016/08/02 Javascript
jquery 判断div show的状态实例
2016/12/03 Javascript
JS实现的系统调色板完整实例
2016/12/21 Javascript
js仿搜狐视频记录片列表展示效果
2020/05/30 Javascript
详解使用vue脚手架工具搭建vue-webpack项目
2017/05/10 Javascript
微信小程序实现的贪吃蛇游戏【附源码下载】
2018/01/03 Javascript
JavaScript中EventLoop介绍
2018/01/22 Javascript
[04:03]DOTA2肉山黑名单梦之声 风暴之灵中文配音鉴赏
2013/07/03 DOTA
零基础写python爬虫之使用Scrapy框架编写爬虫
2014/11/07 Python
python爬虫之自动登录与验证码识别
2020/06/15 Python
python多进程读图提取特征存npy
2019/05/21 Python
使用python3调用wxpy模块监控linux日志并定时发送消息给群组或好友
2019/06/05 Python
Django自定义用户登录认证示例代码
2019/06/30 Python
详解python websocket获取实时数据的几种常见链接方式
2019/07/01 Python
python代码实现逻辑回归logistic原理
2019/08/07 Python
python使用 __init__初始化操作简单示例
2019/09/26 Python
python使用Matplotlib改变坐标轴的默认位置
2019/10/18 Python
Python包,__init__.py功能与用法分析
2020/01/07 Python
HTML5中使用json对象的实例代码
2018/09/10 HTML / CSS
Currentbody澳大利亚:美容仪专家
2019/11/11 全球购物
仓库规划计划书
2014/04/28 职场文书
接收函格式
2015/01/30 职场文书
温馨祝福晨语:美丽的一天从我的问候开始
2019/11/28 职场文书