2017-07-06 13 views
1

KerasモデルをCNTKの形式に保存して、CNTKのAPIを使用して読み込む方法はありますか?KerTモデルをCNTK形式で保存する方法

私は以下のコードを試してみましたが、それは失敗しました:あなたは詳細hereを見つけるでしょう

loaded_model = cntk.load('model2.dnn')

:あなたのコードの後

a = Input(shape=(224,224,3)) 
b = Conv2D(64, (2,2), name='conv1')(a) 
model = Model(inputs=a, outputs=b) 
cntk.combine(model.outputs).save('model2.dnn') 

答えて

0

を、これを試してみてください。

0

(バックエンドとしてCNTKを使用した場合)CNTK形式でKerasモデルを保存するにはKeras-CNTK saving model-v2 format

を参照してください。

import cntk as C 
from keras.backend import load_model 

keras_model = load_model('my_keras_model.h5') 

cntk.combine(keras_model.model.outputs).save('my_cntk_model') 
cntk_model = C.load_model('my_cntk_model') 

(バックエンドとしてTensorFlowを使用した場合)TensorFlow形式でKerasを保存するには:

model = load_model('model2.h5') 
sess = K.get_session() 
tf_saver = tf.train.Saver() 
tf_saver.save(sess=sess, save_path=checkpoint_path) 
+0

はい。私はモデルを保存することができます。しかし、保存されたモデルファイルをロードしようとすると、エラーを報告します: 'raise ValueError( 'ファイルでもバイトバッファでもないモデルをロードできません')RuntimeError:SWIGディレクターメソッドのエラー' – logen

関連する問題