python - 训练好的 Keras 模型无法使用 load_model 加载

Question

我已经用 Tensorflow 后端训练了 Keras 模型。它是用 model.save 保存的。我现在想使用 model_load 重新加载模型，但是，我收到以下错误:

Traceback (most recent call last):

  File "<ipython-input-235-387752c910a4>", line 1, in <module>
    load_model('MyModel.h5')

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\models.py", line 243, in load_model
    model = model_from_config(model_config, custom_objects=custom_objects)

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\models.py", line 317, in model_from_config
    return layer_module.deserialize(config, custom_objects=custom_objects)

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\layers\__init__.py", line 55, in deserialize
    printable_module_name='layer')

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\utils\generic_utils.py", line 144, in deserialize_keras_object
    list(custom_objects.items())))

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\engine\topology.py", line 2514, in from_config
    process_layer(layer_data)

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\engine\topology.py", line 2500, in process_layer
    custom_objects=custom_objects)

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\layers\__init__.py", line 55, in deserialize
    printable_module_name='layer')

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\utils\generic_utils.py", line 144, in deserialize_keras_object
    list(custom_objects.items())))

  File "C:\Anaconda\envs\tensorflow\lib\site-packages\keras\models.py", line 1367, in from_config
    if 'class_name' not in config[0] or config[0]['class_name'] == 'Merge':

KeyError: 0

据我所知，当使用旧版本 Keras 训练的模型加载最新版本时，Keras 中似乎存在错误。所以可能存在版本不匹配的情况。但是，我找不到符合我情况的报告。不能选择降级 Keras 或重新训练。

有人遇到过这个问题，甚至找到了解决方案吗？我将非常感激!

谢谢!

Answer 1

供将来引用:这是配置文件中的问题。 Keras 2.2.4 对此进行了修复:

Keras 2.2.4

@fchollet fchollet 于 10 月 3 日发布了此版本 · 自此发布以来已有 79 人提交到 master Assets 2

这是一个错误修复版本，解决了两个问题:

能够在同名文件已存在时保存模型。加载顺序模型的旧配置文件时出现问题。

因此，我最终使用最新的 TF 和 Keras 版本创建了一个新的虚拟环境。