繁体   English   中英

无法加载 Keras 保存的 model(错误:无法打开文件)

[英]Unable to load a Keras saved model (Error: unable to open file)

我在 colab 上训练了 keras 顺序 model 并将其保存为 TensorFlow SavedModel 格式。 然后我将经过训练的 model(包含 saved_model.pb 文件和变量子文件夹)的压缩文件夹下载到我的计算机上,并尝试将此 model 加载到我的一个 python 文件中
此错误:(注意:我从 anaconda 自定义环境运行我的 python 文件,同时安装了 keras 和 Z2C39BC19B761AC36DC6DFE4474)

OSError: Unable to open file (unable to open file: name = 'C:/Users/user/Desktop/chiffre.model', errno = 13, error message = 'Permission denied', flags = 0, o_flags = 0)

这是colab上的训练和保存代码:

import numpy as np
from tensorflow import keras
from tensorflow.keras import layers

model = keras.Sequential(..................)
#some lines of code
model.save('chiffre.model')

我尝试在 python 文件中加载 model :

new_model = keras.models.load_model('C:/Users/user/Desktop/chiffre.model',custom_objects=None,compile=True)

像这样试试

model.save("my_model")

调用model.save('my_model')会创建一个名为 my_model 的文件夹,其中包含以下内容:

assets  saved_model.pb  variables

model 架构和训练配置(包括优化器、损失和指标)存储在 saved_model.pb 中。 权重保存在 variables/ 目录中。

然后加载 model,

reconstructed_model = keras.models.load_model("my_model")



您还可以保存一个包含模型架构、权重值和 compile() 信息的 HDF5 文件。 它是 SavedModel 的轻量级替代品。

# Calling `save('my_model.h5')` creates a h5 file `my_model.h5`.
model.save("my_h5_model.h5")

要使用此方法加载,

# It can be used to reconstruct the model identically.
reconstructed_model = keras.models.load_model("my_h5_model.h5")

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM