首页 > 解决方案 > KeyError:“class_name”正在加载 Keras json 文件

问题描述

我已经在 imagenet 上为预训练的 resten50 下载了 json 和 h5 文件。但是,加载 json 文件会给我带来 keyerror。

我的代码:

json_file = open('models/resnet50/imagenet_class_index_new.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json)

我收到以下错误。

> Traceback (most recent call last):
  File "/home/mycom/keras-flask-deploy-webapp/app.py", line 155, in <module>

    loaded_model = model_from_json(loaded_model_json)

  File "/home/mycom/.local/lib/python3.5/site-packages/keras/models.py", line 213, in model_from_json

    return layer_from_config(config, custom_objects=custom_objects)
  File "/home/mycom/.local/lib/python3.5/site-packages/keras/utils/layer_utils.py", line 27, in layer_from_config

    class_name = config['class_name']

KeyError: 'class_name'

可能是什么问题?我恢复到 keras==1.2.2 但问题仍然存在。

标签: jsonkerasresnetimagenet

解决方案


推荐阅读