summaryrefslogtreecommitdiff
path: root/source/use_case
diff options
context:
space:
mode:
authorRichard Burton <richard.burton@arm.com>2022-05-17 12:52:50 +0100
committerRichard Burton <richard.burton@arm.com>2022-05-20 11:08:32 +0100
commit6f6df0934f991b64fef494b86643b3f5074fca0e (patch)
tree3f04c50c7ee3bcaea4fa4e9c64c81b27d7cfc4fa /source/use_case
parent8c61c0a3cb8d6b534d1e423211e06b89f45bf223 (diff)
downloadml-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')
-rw-r--r--source/use_case/object_detection/src/UseCaseHandler.cc9
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();