我正在尝试使用 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 ×1