2017-09-18 16 views
0

CNNモデルのウェイトファイル(.h5)のURLがあります。私はその重量を直接pythonファイルにロードし、kerasモデルをコンパイルしたいと思います。どのようにこれを行うことができますか? ダイレクトアプローチがありますか、またはウェイトファイルをダウンロードしてディスクからロードする必要がありますか?URLからウェイトをロード

答えて

2

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を使用できます。 しかし、すべての場合、重みはコンピュータに保存されます。

+0

私はリクエストライブラリで重量をロードしました。今度はkeras get_fileを試してみます。 –

関連する問題