tensorflow(tf)keras save h5、hdf5模型,loadmodel现AttributeError: ‘str‘ object has no attribute ‘decode‘

时间:4年前   阅读:15225

1. 问题说明

keras保存的h5、hdf5模型,在使用load_model和load_weights函数,出现AttributeError: ‘str’ object has no attribute 'decode’问题,之前也使用过这些模型和加载函数,但并没有出现这个错误,找了好久才找到解决方案。

2. 解决办法

貌似是由于h5py模块的版本问题,改用2.10版本

pip install h5py==2.10

3. h5,hdf5模型格式说明

1.HDF5格式文件保存的只是权值文件,需要先构建模型;
2.H5 格式文件保存的是模型的权值和模型的结构,可以直接读取。

network.save("./h5/network.h5") #network为一个训练过的model
networkh5 = keras.models.load_model("./h5/network.h5") #networkh5为一个新建的model,导入network.h5


版权声明:本文为期权记的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。

原文链接:https://www.qiquanji.com/post/9667.html

微信扫码关注

更新实时通知

上一篇:国家经济内循环

下一篇:(tensorflow)tf.keras.callbacks.ModelCheckpoint在训练期间保存模型

网友评论

请先 登录 再评论,若不是会员请先 注册