keras 在调用load_weights的方法是报 ‘str‘ object has no attribute ‘decode‘

最新推荐文章于 2023-04-11 16:10:27 发布

天马行空波

最新推荐文章于 2023-04-11 16:10:27 发布

阅读量973

点赞数 1

分类专栏： python

原文链接：https://blog.csdn.net/xiqi4145/article/details/109547242

版权

python 专栏收录该内容

29 篇文章 0 订阅

订阅专栏

在重新部署模型时遇到问题，Keras在尝试加载权重时出现AttributeError，原因是h5py版本不兼容。解决方案是卸载现有h5py模块并安装版本为2.10的h5py。此问题强调了在编写Dockerfile时指定依赖包版本的重要性，以避免因版本升级导致的未知错误。

摘要由CSDN通过智能技术生成

之前的项目运行的很正常的，在切换了一次模型重新部署的时候发现构建成功，运行时会报错。

keras 在调用load_weights的方法是报错

  File "/usr/local/lib/python3.6/site-packages/keras/engine/network.py", line 1166, in load_weights
    f, self.layers, reshape=reshape)
  File "/usr/local/lib/python3.6/site-packages/keras/engine/saving.py", line 1004, in load_weights_from_hdf5_group
    original_keras_version = f.attrs['keras_version'].decode('utf8')
AttributeError: 'str' object has no attribute 'decode'

解决办法：
卸载原来的h5py模块，安装2.10版本

pip install h5py==2.10

警示：
在写dockerfile的时候一定要把每个包的版本指定为当前环境的版本，不然后续版本更新可能会引发很奇怪的bug导致浪费时间
原文链接：https://blog.csdn.net/xiqi4145/article/details/109547242