我有CNN模型的权重文件(.h5
)的URL。我想将该权重直接加载到Python文件并编译一个keras模型。怎么办呢。
是否有直接方法,或者我应该下载重量文件并从磁盘加载。
答案 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。 但在所有情况下,权重都将保存在您的计算机上。