diff options
author | Richard Burton <richard.burton@arm.com> | 2022-05-17 12:52:50 +0100 |
---|---|---|
committer | Richard Burton <richard.burton@arm.com> | 2022-05-20 11:08:32 +0100 |
commit | 6f6df0934f991b64fef494b86643b3f5074fca0e (patch) | |
tree | 3f04c50c7ee3bcaea4fa4e9c64c81b27d7cfc4fa /tests/use_case/object_detection/InferenceTestYoloFastest.cc | |
parent | 8c61c0a3cb8d6b534d1e423211e06b89f45bf223 (diff) | |
download | ml-embedded-evaluation-kit-6f6df0934f991b64fef494b86643b3f5074fca0e.tar.gz |
Remove dependency on extern defined constants from OD use case
OD API now takes in these paramaters as part of the constructor
Change-Id: I4cce25e364b2a99847b4540440db059997f6a81b
Diffstat (limited to 'tests/use_case/object_detection/InferenceTestYoloFastest.cc')
-rw-r--r-- | tests/use_case/object_detection/InferenceTestYoloFastest.cc | 6 |
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; |