tensorflow-gpu1.4.0运行keras-yolov3报错（keras版本问题）

wechat-920086481

于 2020-08-20 16:34:44 发布

阅读量581

点赞数

CC 4.0 BY-SA版权

分类专栏： tool 文章标签：深度学习

本文链接：https://blog.csdn.net/zahidzqj/article/details/108126548

tool 专栏收录该内容

8 篇文章

订阅专栏

本文记录了在Windows 10环境下运行Keras-YOLOv3遇到的导入错误及解决方案，涉及keras版本与tensorflow版本的适配问题。通过调整导入方式和安装特定版本的keras(2.1.4)和tensorflow-gpu(1.4.0)，成功解决了ImportError和TypeError异常，使程序得以顺利执行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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系统。