diff options
Diffstat (limited to 'tests/use_case/object_detection/InferenceTestYoloFastest.cc')
-rw-r--r-- | tests/use_case/object_detection/InferenceTestYoloFastest.cc | 9 |
1 files changed, 2 insertions, 7 deletions
diff --git a/tests/use_case/object_detection/InferenceTestYoloFastest.cc b/tests/use_case/object_detection/InferenceTestYoloFastest.cc index 8ef012d..2c035e7 100644 --- a/tests/use_case/object_detection/InferenceTestYoloFastest.cc +++ b/tests/use_case/object_detection/InferenceTestYoloFastest.cc @@ -94,13 +94,8 @@ void TestInferenceDetectionResults(int imageIdx, arm::app::Model& model, T toler REQUIRE(tflite::GetTensorData<T>(output_arr[i])); } - arm::app::object_detection::DetectorPostprocessing postp; - postp.RunPostProcessing( - nRows, - nCols, - output_arr[0], - output_arr[1], - results); + arm::app::DetectorPostProcess postp{output_arr[0], output_arr[1], results, nRows, nCols}; + postp.DoPostProcess(); std::vector<std::vector<arm::app::object_detection::DetectionResult>> expected_results; GetExpectedResults(expected_results); |