diff options
author | Isabella Gottardi <isabella.gottardi@arm.com> | 2022-01-27 16:39:37 +0000 |
---|---|---|
committer | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2022-02-08 16:32:28 +0000 |
commit | 3107aa2152de9be8317e62da1d0327bcad6552e2 (patch) | |
tree | 2ba12a5dd39f28ae1b646e132fbe575c6a442ee9 /tests/use_case/object_detection/ObjectDetectionUCTest.cc | |
parent | 5cdfa9b834dc5a94c70f9f2b1f5c849dc5439e85 (diff) | |
download | ml-embedded-evaluation-kit-3107aa2152de9be8317e62da1d0327bcad6552e2.tar.gz |
MLECO-2873: Object detection usecase follow-up
Change-Id: Ic14e93a50fb7b3f3cfd9497bac1280794cc0fc15
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Diffstat (limited to 'tests/use_case/object_detection/ObjectDetectionUCTest.cc')
-rw-r--r-- | tests/use_case/object_detection/ObjectDetectionUCTest.cc | 14 |
1 files changed, 5 insertions, 9 deletions
diff --git a/tests/use_case/object_detection/ObjectDetectionUCTest.cc b/tests/use_case/object_detection/ObjectDetectionUCTest.cc index 0a0486e..2e63f36 100644 --- a/tests/use_case/object_detection/ObjectDetectionUCTest.cc +++ b/tests/use_case/object_detection/ObjectDetectionUCTest.cc @@ -15,7 +15,7 @@ * limitations under the License. */ #include "DetectionResult.hpp" -//include "Detector.hpp" +#include "DetectorPostProcessing.hpp" #include "hal.h" #include "YoloFastestModel.hpp" #include "UseCaseHandler.hpp" @@ -25,7 +25,6 @@ TEST_CASE("Model info") { -printf("Entering Model info \n"); /* Model wrapper object. */ arm::app::YoloFastestModel model; @@ -43,7 +42,6 @@ printf("Entering Model info \n"); TEST_CASE("Inference by index") { -printf("Entering Inference by index \n"); hal_platform platform; data_acq_module data_acq; data_psn_module data_psn; @@ -67,17 +65,15 @@ printf("Entering Inference by index \n"); caseContext.Set<hal_platform&>("platform", platform); caseContext.Set<arm::app::Model&>("model", model); caseContext.Set<uint32_t>("imgIndex", 0); + arm::app::object_detection::DetectorPostprocessing postp; + caseContext.Set<arm::app::object_detection::DetectorPostprocessing&>("postprocess", postp); REQUIRE(arm::app::ObjectDetectionHandler(caseContext, 0, false)); - - auto results = caseContext.Get<std::vector<arm::app::DetectionResult>>("results"); - } TEST_CASE("Inference run all images") { - printf("Entering Inference run all images \n"); hal_platform platform; data_acq_module data_acq; data_psn_module data_psn; @@ -101,15 +97,15 @@ TEST_CASE("Inference run all images") caseContext.Set<hal_platform&>("platform", platform); caseContext.Set<arm::app::Model&>("model", model); caseContext.Set<uint32_t>("imgIndex", 0); + arm::app::object_detection::DetectorPostprocessing postp; + caseContext.Set<arm::app::object_detection::DetectorPostprocessing&>("postprocess", postp); REQUIRE(arm::app::ObjectDetectionHandler(caseContext, 0, true)); - } TEST_CASE("List all images") { -printf("Entering List all images \n"); hal_platform platform; data_acq_module data_acq; data_psn_module data_psn; |