mask rcnn 加载权重报错

mask rcnn 加载权重报错

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 -i https://pypi.tuna.tsinghua.edu.cn/simple/

仍然不行,那只有从报错的源码入手了
根据报错提示问题出在
xxx\Lib\site-packages\keras\engine\topology.py
这个文件的3418,3422行。
python3中,编码的时候区分了字符串和二进制
python3在bytes和str两种类型转换,所需要的函数依次是encode(),decode()
这里f.attrs[‘keras_version’]类型我们不知道,因此需要判断是str还是bytes以便使用相应的方法
如果不判断,这个类型一会是str,一会是bytes,这个bug让我找了好久
在这里插入图片描述
代码如下

    if 'keras_version' in f.attrs:
        original_keras_version = f.attrs['keras_version'].encode('utf-8') if isinstance(f.attrs['keras_version'],str) else f.attrs['keras_version'].decode('utf-8')
    else:
        original_keras_version = '1'
    if 'backend' in f.attrs:
        original_backend = f.attrs['backend'].encode('utf-8') if isinstance(f.attrs['backend'],str) else f.attrs['backend'].decode('utf-8')
    else:
        original_backend = None

``
正常运行截图
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值