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 /source/use_case/object_detection/src | |
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 'source/use_case/object_detection/src')
-rw-r--r-- | source/use_case/object_detection/src/UseCaseHandler.cc | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/source/use_case/object_detection/src/UseCaseHandler.cc b/source/use_case/object_detection/src/UseCaseHandler.cc index e9bcd4a..a7acb46 100644 --- a/source/use_case/object_detection/src/UseCaseHandler.cc +++ b/source/use_case/object_detection/src/UseCaseHandler.cc @@ -27,9 +27,6 @@ namespace arm { namespace app { - namespace object_detection { - extern const int channelsImageDisplayed; - } /* namespace object_detection */ /** * @brief Presents inference results along using the data presentation @@ -102,8 +99,12 @@ namespace app { DetectorPreProcess preProcess = DetectorPreProcess(inputTensor, true, model.IsDataSigned()); std::vector<object_detection::DetectionResult> results; + const object_detection::PostProcessParams postProcessParams { + inputImgRows, inputImgCols, object_detection::originalImageSize, + object_detection::anchor1, object_detection::anchor2 + }; DetectorPostProcess postProcess = DetectorPostProcess(outputTensor0, outputTensor1, - results, inputImgRows, inputImgCols); + results, postProcessParams); do { /* Ensure there are no results leftover from previous inference when running all. */ results.clear(); |