machine-learning - 从 URL 加载权重

Question

我有 CNN 模型的权重文件 (.h5) 的 URL。我想将该权重直接加载到 Python 文件并编译 keras 模型。这怎么办？

是否有直接的方法，或者我应该下载权重文件并从磁盘加载它？

Answer 1

有一个功能可以从 URL 下载权重并将其存储在 ~/.keras/

这是函数 get_file

from keras.utils.data_utils import get_file

示例:

from keras.utils.data_utils import get_file
weights_path = get_file(
            'the name under the model will be saved',
            'YOUR URL')
model.load_weights(weights_path)

它返回保存权重的路径。然后你可以使用load_weight。但在所有情况下，权重都会保存在您的计算机上。