diff options
| -rw-r--r-- | tracker-neuralnet/ftnoir_tracker_neuralnet.cpp | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/tracker-neuralnet/ftnoir_tracker_neuralnet.cpp b/tracker-neuralnet/ftnoir_tracker_neuralnet.cpp index 3687a6cd..50d1e7b4 100644 --- a/tracker-neuralnet/ftnoir_tracker_neuralnet.cpp +++ b/tracker-neuralnet/ftnoir_tracker_neuralnet.cpp @@ -315,7 +315,16 @@ std::optional<PoseEstimator::Face> PoseEstimator::run( const auto nt = omp_get_num_threads(); omp_set_num_threads(num_threads); - session.Run(Ort::RunOptions{nullptr}, input_names, &input_val, 1, output_names, output_val, 3); + try + { + session.Run(Ort::RunOptions{ nullptr }, input_names, &input_val, 1, output_names, output_val, 3); + } + catch (const Ort::Exception &e) + { + qDebug() << "Failed to run the model: " << e.what(); + omp_set_num_threads(nt); + return {}; + } omp_set_num_threads(nt); // FIXME: Execution time fluctuates wildly. 19 to 26 ms. Why??? |