python - 如何使用 C_API 导入训练有素的 tf.contrib.learn.dnnclassifier

Question

如何使用 C_API 导入经过训练的 DNNClassifer？我尝试使用 TF_LoadSessionFromSavedModel，但它引发了以下错误：

我 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);

Answer 1

我最近遇到了同样的错误。真正帮助我的是看一下TF_Message(s)（这个表达式应该返回一个以 null 结尾的 c 字符串（即 char*））指向的内容。