python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > tf.keras.models.load_model加载模型报错

解决tf.keras.models.load_model加载模型报错问题

作者:_不负时光

这篇文章主要介绍了解决tf.keras.models.load_model加载模型报错问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教

tf.keras.models.load_model 加载模型报错记录

Unknown activation function:leaky_relu

解决办法

是在tf.keras.models.load_model()括号里面加上         

custom_objects = {“YYY”:XXX对应的函数或者对象}

例如报错 

ValueError: Unknown activation function:leaky_relu 

就加上

new_model=tf.keras.models.load_model("models/model.h5", custom_objects = {"leaky_relu":tf.nn.leaky_relu})

keras.models.load_model无法打开h5文件

是个小萌新,用tensorflow.keras.models.load_model打开h5文件时总是失败,各位大佬的解决方法过于深奥,实在看不懂,最后自己用下面办法解决的

1.可能是h5py与tensorflow版本不匹配,试试使用h5py2.10版本

2.版本调整后:

from tensorflow import keras
model=keras.models.load_model('model.h5')

运行代码仍然出现以下问题:

更改代码如下:

from tensorflow import keras
import h5py
import os 
file=h5py.File('model.h5')
model=keras.models.load_model(file)

试试打印信息:

成功了

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文