当前位置:  开发笔记 > 编程语言 > 正文

如何用c ++语言加载tensorflow.so和c_api.h的图形?

如何解决《如何用c++语言加载tensorflow.so和c_api.h的图形?》经验,为你挑选了1个好方法。

我无法找到有关如何加载图形与任何例子tensorflow.soc_api.hC++中.我读了c_api.h,但ReadBinaryProto功能不在其中.如何在没有该ReadBinaryProto功能的情况下加载图形?



1> ash..:

如果您使用的是C++,则可能需要使用C++ API.该标签图像例如可能会是一个很好的样本,以帮助您开始.

如果您确实只想使用C API,请使用TF_GraphImportGraphDef加载图表.请注意,C API使用起来不是特别方便(它打算用其他语言构建绑定,例如Go,Java,Rust,Haskell等)例如:

#include                                                                         
#include                                                                        
#include                                                            

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


谢谢您的回答。我想在tensorflow项目之外使用C ++ API。``标签图像示例''构建在tensorflow项目中。我如何在tensorflow之外使用C ++ API?我应该包含哪些头文件?
使用C Api的@ash可能是不好的,但是不幸的是,这是在目标系统上运行推理而无需安装完整的tensorflow且不必使用pip的唯一方法。提前编译也是另一种方法,但是它仍然不支持很多模块,并且文档几乎不存在。
推荐阅读
php
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有