python - 尝试将模型加载到另一个平台时 Keras ValueError : Unknown layer:name,

Question

我在 Nvidia Quadro 板上使用 Keras 2.2.4 训练了一个卷积神经网络。我将经过训练的模型保存在两个单独的文件中:一个文件 (model.json) 描述架构，另一个文件 (model.h5) 包含所有权重。

我想将保存的模型加载到运行 Keras 2.2.2 的 Nvidia Jetson TX2 板上，我正在尝试按如下方式进行:

# load json and create model
    json_file = open(prefix+'final_model.json', 'r')
    loaded_model_json = json_file.read()
    json_file.close()
    loaded_model = model_from_json(loaded_model_json)
# load weights into new model
    loaded_model.load_weights(prefix+"model.h5")
    model = loaded_model

但是，当我尝试加载时出现以下错误:

loaded_model = model_from_json(loaded_model_json)

文件“/home/nvidia/.local/lib/python3.5/site-packages/keras/engine/saving.py”，第 368 行，在 model_from_json 中 返回反序列化(配置，custom_objects=custom_objects) 反序列化中的文件“/home/nvidia/.local/lib/python3.5/site-packages/keras/layers/init.py”，第 55 行 printable_module_name='层') 文件“/home/nvidia/.local/lib/python3.5/site-packages/keras/utils/generic_utils.py”，第 145 行，在 deserialize_keras_object 中 列表(custom_objects.items()))) 文件“/home/nvidia/.local/lib/python3.5/site-packages/keras/engine/sequential.py”，第 292 行，在 from_config 自定义对象=自定义对象) 反序列化中的文件“/home/nvidia/.local/lib/python3.5/site-packages/keras/layers/init.py”，第 55 行 printable_module_name='层') 文件“/home/nvidia/.local/lib/python3.5/site-packages/keras/utils/generic_utils.py”，第 165 行，在 deserialize_keras_object 中 ':' + 函数名) ValueError: Unknown layer:name

我也试过将整个模型保存在一个文件中，但得到了同样的错误。

我已经尝试了 here 中的解决方案但无法解决它。

有没有人见过这个错误？有什么建议吗？

Answer 1

我昨天遇到了同样的问题，我刚刚通过 conda 更新了 keras，一切正常。