Function LoadGraph

examples/tensorflow_wrapper/label_image/label_image.cpp:109–137

}

static TF_Session* LoadGraph(const char* model_file, TF_Graph* graph)
{
    TF_Status* s = TF_NewStatus();

    vector<char> model_buf;

    if(ReadEntireFile(model_file, model_buf) < 0)
        return nullptr;

    TF_Buffer graph_def = {model_buf.data(), model_buf.size(), nullptr};

    TF_ImportGraphDefOptions* import_opts = TF_NewImportGraphDefOptions();
    TF_ImportGraphDefOptionsSetPrefix(import_opts, "");
    TF_GraphImportGraphDef(graph, &graph_def, import_opts, s);

    if(TF_GetCode(s) != TF_OK)
    {
        printf("load graph failed!\n Error: %s\n", TF_Message(s));
        return nullptr;
    }

    TF_SessionOptions* sess_opts = TF_NewSessionOptions();
    TF_Session* session = TF_NewSession(graph, sess_opts, s);
    assert(TF_GetCode(s) == TF_OK);

    TF_DeleteStatus(s);

    return session;
}

static int PrintTopLabels(const vector<TF_Tensor*>& outputs, const char* labels_file)
{