如何修复“名称错误：名称'load_model'未定义”

Question

我正在阅读《Python深度学习》一书，我遇到了一个问题。我已经使用 Pycharm 将模型保存在 anther 文件中，当我尝试通过函数“load_model()”加载它时，系统支持“NameError：名称“load_model”未定义”。pycharm 2019.1.3 keras 2.2.4 张量流 1.13.0

from keras import models
model = load_model('cats_and_dogs_small_2.h5')

使用 TensorFlow 后端。回溯（最近一次调用最后）：文件“F：/ python程序/ VisualizDemo/main.py”，第2行，模型= load_model（'cats_and_dogs_small_2.h5'）NameError：名称'load_model'未定义

进程已完成，退出代码为 1

Answer 1

这里缺少的是， load_model 函数位于模型类内部，因此您必须引用模型类。

from keras import models    
model = models.load_model('filename.h5')

为了按照您的方式进行，您必须使用以下导入

from keras.models import load_model

PS：下一行可能会对您将来有所帮助。如果您尝试加载权重，请使用函数：

model.load_weight('weights_file.h5')