我正在从.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