在C++中,我找不到任何关于如何使用tensorflow.so和c_api.h加载图形的示例。我读了c_api.h,但是ReadBinaryProto函数不在其中。如果没有ReadBinaryProto函数,我如何加载图形?
tensorflow.so
c_api.h
ReadBinaryProto
pcww981p1#
如果你使用的是C++,你可能会想用C++ API来代替,label image example可能是一个很好的例子来帮助你开始。如果你真的只想使用C API,可以使用TF_GraphImportGraphDef来加载一个图形。注意C API使用起来并不特别方便(它的目的是在其他语言中构建绑定,如Go,Java,Rust,Haskell等)例如:
TF_GraphImportGraphDef
#include <stdio.h> #include <stdlib.h> #include <tensorflow/c/c_api.h> TF_Buffer* read_file(const char* file); void free_buffer(void* data, size_t length) { free(data); } int main() { // Graph definition from unzipped https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip // which is used in the Go, Java and Android examples TF_Buffer* graph_def = read_file("tensorflow_inception_graph.pb"); TF_Graph* graph = TF_NewGraph(); // Import graph_def into graph TF_Status* status = TF_NewStatus(); TF_ImportGraphDefOptions* opts = TF_NewImportGraphDefOptions(); TF_GraphImportGraphDef(graph, graph_def, opts, status); TF_DeleteImportGraphDefOptions(opts); if (TF_GetCode(status) != TF_OK) { fprintf(stderr, "ERROR: Unable to import graph %s", TF_Message(status)); return 1; } fprintf(stdout, "Successfully imported graph"); TF_DeleteStatus(status); TF_DeleteBuffer(graph_def); // Use the graph TF_DeleteGraph(graph); return 0; } TF_Buffer* read_file(const char* file) { FILE *f = fopen(file, "rb"); fseek(f, 0, SEEK_END); long fsize = ftell(f); fseek(f, 0, SEEK_SET); //same as rewind(f); void* data = malloc(fsize); fread(data, fsize, 1, f); fclose(f); TF_Buffer* buf = TF_NewBuffer(); buf->data = data; buf->length = fsize; buf->data_deallocator = free_buffer; return buf; }
sycxhyv72#
如果您希望在TensorFlow项目之外使用它(因此不使用Bazel构建),那么前面的答案是您的主要选择。您需要使用TF_GraphImportDef从c_api. h加载它,我建议您使用Python进行培训和测试,然后在完成后导出模型/图形以供C++/C Api使用。
2条答案
按热度按时间pcww981p1#
如果你使用的是C++,你可能会想用C++ API来代替,label image example可能是一个很好的例子来帮助你开始。
如果你真的只想使用C API,可以使用
TF_GraphImportGraphDef
来加载一个图形。注意C API使用起来并不特别方便(它的目的是在其他语言中构建绑定,如Go,Java,Rust,Haskell等)例如:sycxhyv72#
如果您希望在TensorFlow项目之外使用它(因此不使用Bazel构建),那么前面的答案是您的主要选择。您需要使用TF_GraphImportDef从c_api. h加载它,我建议您使用Python进行培训和测试,然后在完成后导出模型/图形以供C++/C Api使用。