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
微信扫码关注
更新实时通知
网友评论