def export_model_froms(output_node_names): tf.train.Saver().save(K.get_session(), 'out/' + TargetModelName + '.chkp') freeze_graph('out/', output_graph=TargetModelName + "_model.pb", output_node_names=output_node_names)
def export_model_froms(output_node_names): ''' 将 session 写到文件 out/googlenet.chkp.*** .meta文件保存了当前图结构;.index文件保存了当前参数名;.data文件保存了当前参数值。 ''' tf.train.Saver().save(K.get_session(), 'out/' + TargetModelName + '.chkp') freeze_graph('out/', output_graph=TargetModelName+"_model.pb", output_node_names=output_node_names)