keras - 如何以 CNTK 格式保存 Keras 模型

Question

有没有办法将 Keras 模型保存为 CNTK 的格式，以便我们可以使用 CNTK 的 API 加载它？

我已经尝试过下面的代码，但失败了:

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')

Answer 1

请查看 Keras-CNTK saving model-v2 format

以 CNTK 格式保存 Keras 模型(使用 CNTK 作为后端时):

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 格式保存 Keras(使用 TensorFlow 作为后端时):

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