summaryrefslogtreecommitdiff
path: root/tests/use_case/object_detection/InferenceTestYoloFastest.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tests/use_case/object_detection/InferenceTestYoloFastest.cc')
-rw-r--r--tests/use_case/object_detection/InferenceTestYoloFastest.cc6
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/use_case/object_detection/InferenceTestYoloFastest.cc b/tests/use_case/object_detection/InferenceTestYoloFastest.cc
index eb92904..d328684 100644
--- a/tests/use_case/object_detection/InferenceTestYoloFastest.cc
+++ b/tests/use_case/object_detection/InferenceTestYoloFastest.cc
@@ -104,7 +104,11 @@ void TestInferenceDetectionResults(int imageIdx, arm::app::Model& model, T toler
REQUIRE(tflite::GetTensorData<T>(output_arr[i]));
}
- arm::app::DetectorPostProcess postp{output_arr[0], output_arr[1], results, nRows, nCols};
+ const arm::app::object_detection::PostProcessParams postProcessParams {
+ nRows, nCols, arm::app::object_detection::originalImageSize,
+ arm::app::object_detection::anchor1, arm::app::object_detection::anchor2
+ };
+ arm::app::DetectorPostProcess postp{output_arr[0], output_arr[1], results, postProcessParams};
postp.DoPostProcess();
std::vector<std::vector<arm::app::object_detection::DetectionResult>> expected_results;