TensorFlow使用C++加载使用训练好的模型，.cc文件代码实现的相关类及方法总结

最新推荐文章于 2023-11-03 17:50:44 发布

IceBear6

最新推荐文章于 2023-11-03 17:50:44 发布

阅读量3.1k

点赞数

分类专栏： Machine Learning

本文链接：https://blog.csdn.net/BadMushroom/article/details/78720582

版权

本文总结了使用TensorFlow C++ API加载预训练模型的关键步骤，包括整体逻辑、Run函数的应用、NewSession的创建、ReadBinaryProto的模型读取以及Tensor的各种操作如vec、matrix和tensor函数。参考了相关教程，详细阐述了.cc文件中的实现细节。

摘要由CSDN通过智能技术生成

在官网API和Tensorflow源码头文件中查看获取。

同时参考

https://medium.com/jim-fleming/loading-a-tensorflow-graph-with-the-c-api-4caaff88463f

https://vimsky.com/article/3600.html
1.整体逻辑

/// ```c++ /// tensorflow::GraphDef graph; /// // ... Create or load graph into "graph". /// /// // This example uses the default options which connects /// // to a local runtime. /// tensorflow::SessionOptions options; /// std::unique_ptr<tensorflow::Session> /// session(tensorflow::NewSession(options)); /// /// // Create the session with this graph. /// tensorflow::Status s = session->Create(graph); /// if (!s.ok()) { ... } /// /// // Run the graph and fetch the first output of the "output" /// // operation, and also run to but do not return anything /// // for the "update_state" operation. /// std::vector<tensorflow::Tensor> outputs; /// s = session->Run({}, {"output:0"}, {"update_state"}, &outputs); /// if (!s.ok()) { ... } /// /// // Map the output as a flattened float tensor, and do something /// // with it. /// auto output_tensor = outputs[0].flat<float>(); /// if (output_tensor(0) > 0.5) { ... } /// /// // Close the session to release the resources associated with /// // this session. /// session->Close();

2.代码