如何tensorflow.so在c++语言中使用www.example.com和c_api.h加载图形?

jxct1oxe  于 2023-03-03  发布在  其他
关注(0)|答案(2)|浏览(108)

在C++中,我找不到任何关于如何使用tensorflow.soc_api.h加载图形的示例。我读了c_api.h,但是ReadBinaryProto函数不在其中。如果没有ReadBinaryProto函数,我如何加载图形?

pcww981p

pcww981p1#

如果你使用的是C++,你可能会想用C++ API来代替,label image example可能是一个很好的例子来帮助你开始。
如果你真的只想使用C API,可以使用TF_GraphImportGraphDef来加载一个图形。注意C API使用起来并不特别方便(它的目的是在其他语言中构建绑定,如Go,Java,Rust,Haskell等)例如:

#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;
}
sycxhyv7

sycxhyv72#

如果您希望在TensorFlow项目之外使用它(因此不使用Bazel构建),那么前面的答案是您的主要选择。您需要使用TF_GraphImportDef从c_api. h加载它,我建议您使用Python进行培训和测试,然后在完成后导出模型/图形以供C++/C Api使用。

相关问题