TensorFlow c++ API 在 session->run() 上崩溃

Question

我正在编写一个 Tensorflow c++ 代码来检查我在服务器上收到的一系列图像是否属于"真"类别或"假"类别。 我已经使用Python训练了模型，并为模型生成了一个.pb文件。

现在，我将模型加载到 c++ 中，并使用以下代码从我从连接到它的向量接收图像：

for (int iLSP = 0; iLSP < LSPs.size(); iLSP++)
{
Mat image1 []= {LSPs[iLSP], LSPs[iLSP], LSPs[iLSP]};
Mat image;
merge(image1,3,image);
image.convertTo(image, CV_32FC1);
image = image / 255.0;
resize(image, image, Size(Width, Height));
Tensor image_tensor(DT_FLOAT, TensorShape({1,Width,Height,3}));
StringPiece tmp_data = image_tensor.tensor_data();
memcpy(const_cast<char*>(tmp_data.data()), (image.data), Height * Width * sizeof(float));
Session *sess;
SessionOptions options;
TF_CHECK_OK(NewSession(options, &sess));
GraphDef graph_def;
TF_CHECK_OK(ReadBinaryProto(Env::Default(), "models/model.pb", &graph_def));
TF_CHECK_OK(sess->Create(graph_def));
std::vector<std::pair<string, tensorflow::Tensor>> inputs = {{"x", image_tensor }};
std::vector<tensorflow::Tensor> outputs;                
Status status = sess->Run(inputs, {"y_pred"}, {}, &outputs);
if (!status.ok())
{
cout<<"Error: "<<status.ToString()<<endl;
}
auto output_mapped = outputs[0].tensor<float, 2>();
totalFalse += output_mapped(0);
totalTrue  += output_mapped(1);
sess->Close();
image.release();
}

代码编译并运行。问题是在多次连接到服务器并接收到多个图像序列后，代码崩溃而没有任何错误输出或指示崩溃原因(甚至不是分段错误(。

在每一行上都做了一个"cout"来检查代码崩溃的位置，并且它发生在以下行：

Status status = sess->Run(inputs, {"y_pred"}, {}, &outputs);

有没有为什么要调试 sess->run 内部发生的事情？ 因为代码甚至没有填充状态变量来获得想法！ 还有没有办法从会话运行中引发异常？ 正如我已经提到的，这将是一个服务器代码，至少如果会话>运行不起作用，我可以在保持程序运行并且不会崩溃的同时捕获异常完整服务器

Answer 1

在exit()、_exit()和terminate()函数上设置断点并运行代码。