如何将TensorFlow模型(从.ckpt和.meta加载)保存/导出到Keras模型(到.h5文件)?

时间:2018-09-18 21:28:35

标签: python python-3.x tensorflow keras

我正在从.meta和.ckpt文件加载元图

meta_graph = tf.train.import_meta_graph(meta_file)
with tf.Session() as sess:
    meta_graph.restore(tf.get_default_session(), ckpt_file)
.pb文件中的

图形

with gfile.FastGFile(pb_file,'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
    tf.import_graph_def(graph_def, name='')

current_graph = tf.get_default_graph()

然后我想将它们中的任何一个直接转换为Keras模型或.h5文件。

但是我无法为此找到一个辅助函数。 MetaGraph或Graph都没有针对Keras或.h5的导出实用程序,Keras也没有针对MetaGraph或Graph的导入功能。

请帮助我过桥。

文件可以在这里找到: https://github.com/davidsandberg/facenet#pre-trained-models(VGGFace 2)或 https://drive.google.com/file/d/1EXPBSXwTaqrSC0OhUdXNmKSh9qJUQ55-/view

0 个答案:

没有答案