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