从网址

时间:2017-09-18 15:48:28

标签: machine-learning keras

我有CN​​N模型的权重文件(.h5)的URL。我想将该权重直接加载到Python文件并编译一个keras模型。怎么办呢。

是否有直接方法,或者我应该下载重量文件并从磁盘加载。

1 个答案:

答案 0 :(得分:3)

可以从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。 但在所有情况下,权重都将保存在您的计算机上。