From 7f333ba990d2318858f7e7c0c8e7acf4dc166a68 Mon Sep 17 00:00:00 2001 From: Alex Bezdieniezhnykh Date: Thu, 17 Oct 2024 09:14:15 +0300 Subject: [PATCH] ai_player: show all frames, regardless of inference results --- .gitignore | 2 ++ .../src-opi5/aiengineinferenceopi5.cpp | 36 +++++++++---------- 2 files changed, 19 insertions(+), 19 deletions(-) diff --git a/.gitignore b/.gitignore index 5a4f147..5d2fe2f 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,4 @@ MAVSDK/ ardupilot/ +build/ +.idea \ No newline at end of file diff --git a/misc/rtsp_ai_player/src-opi5/aiengineinferenceopi5.cpp b/misc/rtsp_ai_player/src-opi5/aiengineinferenceopi5.cpp index 54421ab..51ac77d 100644 --- a/misc/rtsp_ai_player/src-opi5/aiengineinferenceopi5.cpp +++ b/misc/rtsp_ai_player/src-opi5/aiengineinferenceopi5.cpp @@ -116,7 +116,7 @@ void AiEngineInferenceOpi5::drawObjects(cv::Mat& image, const object_detect_resu //continue; } - fprintf(stderr, "TUOMAS [%d] prop = %f\n", i, result.prop); + fprintf(stderr, "[%d] prop = %f\n", i, result.prop); int left = result.box.left; int top = result.box.top; @@ -153,25 +153,23 @@ void AiEngineInferenceOpi5::performInferenceSlot(cv::Mat frame) return; } - if (od_results.count > 0) { - AiEngineInferenceResult result; - for (int i = 0; i < od_results.count; i++) { - object_detect_result *det_result = &(od_results.results[i]); - qDebug() << "TUOMAS box:" << det_result->box.top << det_result->box.left << det_result->box.bottom << det_result->box.right; - AiEngineObject object; - object.classId = det_result->cls_id; - object.propability = det_result->prop; - object.rectangle.top = det_result->box.top; - object.rectangle.left = det_result->box.left; - object.rectangle.bottom = det_result->box.bottom; - object.rectangle.right = det_result->box.right; - result.objects.append(object); - } - - drawObjects(scaledFrame, od_results); - result.frame = scaledFrame.clone(); - emit resultsReady(result); + AiEngineInferenceResult result; + for (int i = 0; i < od_results.count; i++) { + object_detect_result *det_result = &(od_results.results[i]); + qDebug() << "____box:" << det_result->box.top << det_result->box.left << det_result->box.bottom << det_result->box.right << "\n"; + AiEngineObject object; + object.classId = det_result->cls_id; + object.propability = det_result->prop; + object.rectangle.top = det_result->box.top; + object.rectangle.left = det_result->box.left; + object.rectangle.bottom = det_result->box.bottom; + object.rectangle.right = det_result->box.right; + result.objects.append(object); } + drawObjects(scaledFrame, od_results); + result.frame = scaledFrame.clone(); + emit resultsReady(result); + mActive = false; }