一、 问题描述

最近在跑yolo v3的多目标跟踪,环境:win10+tensorflow+ python +keras
出现以下错误:
raise TypeError(‘Keyword argument not understood:’, kwarg)
解决Keras加载模型TypeError: (‘Keyword argument not understood:’, u’return_state’)

二、原因分析

由于安装了tensorflow,tensorflow自带了keras,所以在自己安装keras时版本需要与tensorflow内置的一样。不一致的话会出现错误。

三、 解决方法

3.1 用下面三句话直接读取模型,然后输出需要安装的keras版本,重新下载版本

import h5py
f = h5py.File('./model_data/yolo.h5', 'r')
print(f.attrs.get('keras_version'))

3.2 重新下载keras版本

我前面输出的版本为2.2.4版本,所以下载2.2.4版本,问题解决。

pip install keras==2.2.4