如何使用C_API导入经过训练的tf.contrib.learn.dnnclassifier

时间:2018-04-12 14:07:20

标签: python c++ tensorflow c-api

如何使用C_API导入经过培训的DNNClassifer?我尝试使用TF_LoadSessionFromSavedModel,但它引发了以下错误:

I tensorflow / cc / saved_model / loader.cc:284]加载SavedModel:失败。

以下是代码的C部分:

    const char* export_dir = "path_to_trained_model/folder_with_trained_model";
    const char* tags_[1] = {"import"};

    TF_Graph* graph_new = TF_NewGraph();
    TF_Buffer* meta_graph_def= TF_NewBuffer();
    TF_SessionOptions* opt = TF_NewSessionOptions();
    TF_Buffer* run_options = TF_NewBufferFromString("", 0);
    TF_Status* s = TF_NewStatus();

    TF_Session* session1= TF_LoadSessionFromSavedModel(
            opt, run_options,  export_dir,
            tags_, 1,  graph_new,
            meta_graph_def, s);

1 个答案:

答案 0 :(得分:0)

我最近遇到了相同的错误。真正帮助我的是看看 TF_Message(s)(此表达式应返回以空值终止的c字符串(即char *))指向的内容。