如何在Tensorflow中加载下载的H5模型

时间:2020-10-20 14:29:02

标签: machine-learning tensorflow2.0

尝试加载从github下载的模型时收到以下错误,并收到以下错误“ SavedModel文件在以下位置不存在:modelname / {saved_model.pbtxt | saved_model.pb}”。

我使用了tf.keras.models.load_model,但是我相信如果已经保存了模型,就可以使用它。是否有加载以前未保存的外部模型的方法?

2 个答案:

答案 0 :(得分:0)

由于错误提示无法找到 model 来加载它,如果您再次保存它,您可以尝试更改模型名称,如果您正在使用,请确保遵循正确的文件夹结构和this 文档中提到的方法。

答案 1 :(得分:0)

您需要提供模型到函数的完整路径(包括文件名)。

my_model = keras.models.load_model("path/to/my_h5_model.h5")

这个函数同时支持字符串和python Path 对象。