tensorflow-gpu1.4.0运行keras-yolov3报错(keras版本问题)

win10中,直接运行keras-yolov3,先提示keras ImportError: cannot import name 'Add'

找到提示的错误地方:

from keras.layers import  Add, Concatenate

这个应该是和keras版本有关,尝试不卸载重装keras,那就直接更改导包方式:import keras

然后将'Add' 修改为'keras.layers.Add',同样将‘Concatenate’修改为类似的方式,然后运行后,该部分没有问题了。

但是出现其他问题:

File "yolo.py", line 71, in generate
    self.yolo_model = load_model(model_path, compile=False)
TypeError: load_model() got an unexpected keyword argument 'compile'

很明显这个和‘from keras.models import load_model ’有关,那应该还要更改keras的版本。

但是这么多keras版本该装哪一个?试着装了2.0.0,提示升级tensorflow-gpu。

碰巧在其他的环境中遇到了tensorflow-gpu1.4.0和keras2.1.4,重装该版本keras后,程序顺利执行完毕。

这种版本的tensorflow和keras组合同样适用于linux系统。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值