Python freeze_graph Beispiele

Beispiel #1

Datei: TrainDeepSpeech.py Projekt: IbrahimAbdallahElgendy/DNN-Inference-Optimization

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)

Beispiel #2

Datei: TrainInceptionV4.py Projekt: IbrahimAbdallahElgendy/DNN-Inference-Optimization

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)