小编Gui*_*ern的帖子

使用 tensorflow c++ api 运行 session 明显比使用 python 慢

我正在尝试使用 tensorflow c++ api(仅限 CPU)运行 SqueezeDet。我已经冻结了张量流图并从 C++ 加载它。虽然在检测质量方面一切都很好,但性能比 python 慢得多。那可能是什么原因?

简化后,我的代码如下所示:

  int main (int argc, const char * argv[])
  {
    // Initializing graph 
    tensorflow::GraphDef graph_def;
    // Folder in which graph data is located
    string graph_file_name = "Model/graph.pb";
    // Loading graph 
    tensorflow::Status graph_loaded_status =  ReadBinaryProto(tensorflow::Env::Default(), graph_file_name, &graph_def);
    if (!graph_loaded_status.ok())
    {
      cout << graph_loaded_status.ToString() << endl;
      return 1;
    }
    unique_ptr<tensorflow::Session> session_sqdet(tensorflow::NewSession(tensorflow::SessionOptions()));
    tensorflow::Status session_create_status = session_sqdet->Create(graph_def);
    if (!session_create_status.ok())
    {
      cout << "Session create status: fail." << endl;
      return 1;
    }
    while ()
    { …
Run Code Online (Sandbox Code Playgroud)

tensorflow

5
推荐指数
1
解决办法
1428
查看次数

标签 统计

tensorflow ×1