python - 加载在线托管的 TensorFlow (.h5) 模型

Question

我正在尝试使用在线存储的链接 ( https://<url> ) 加载 .h5 模型，但我从 TensorFlow 收到此错误:

MODEL_PATH = 'http://<url>/<model_name>.h5'
# I have replace the <tags> with actual URL and file name
model = load_model(MODEL_PATH)

tensorflow.python.framework.errors_impl.UnimplementedError: File system scheme 'http' not implemented 

.h5 的链接非常好，所以它与 TensorFlow 有关。有人可以帮忙吗

Answer 1

虽然我无法找到直接从 URL 加载模型的方法，但我使用了一个技巧来使用 urllib.request 实现类似的功能。包裹。

import urllib.request

urllib.request.urlretrieve(
        'http://<url>/model.h5', 'model.h5')

上面的行将下载 h5 模型并将其存储在项目的根目录中。然后 Python 可以使用下载的模型:

MODEL_PATH = './model.h5'

备注 :以上代码适用于 Python 3。如果您使用的是 Python 2，请使用以下代码:

import urllib
urllib.urlretrieve('http://<url>/model.h5', 'model.h5')