由於keras不同版本的API有變化,因此在一個keras版本下訓練的模型在另一個keras版本下加載時,可能會出現諸如('Keyword argument not understood:', u'data_format')
等報錯。
通過打開*.h5文件,查看該模型訓練所用環境,再安裝該環境即可解決報錯。
查看Keras Model所用的Keras環境的方法
import h5py
f = h5py.File('Model.h5', 'r')
print(f.attrs.get('keras_version'))
根據輸出的keras版本安裝對應版本的keras即可解決加載問題。