diff options
author | Richard Burton <richard.burton@arm.com> | 2022-02-14 11:55:35 +0000 |
---|---|---|
committer | Richard <richard.burton@arm.com> | 2022-02-14 15:02:26 +0000 |
commit | ed35a6fea4a1604db81c56fc71f7756822fcf212 (patch) | |
tree | f04b7d41ded8b4824978c2feaf120a3b6e1be2fb /source/use_case/img_class/src | |
parent | e2da7ee5e9732ec0d1962b7d74737b8ef5463a9e (diff) | |
download | ml-embedded-evaluation-kit-ed35a6fea4a1604db81c56fc71f7756822fcf212.tar.gz |
MLECO-2874: Move NMS out of the OD use_case
* Add ImageUtils
* Move image related code from UseCaseCommonUtils to ImageUtils
* Move NMS related code to ImageUtils
* Delete test specific ImageUtils and use new ImageUtils
Signed-off-by: Richard Burton <richard.burton@arm.com>
Change-Id: Icbf5dd9c6a941b0126ecdf69a0c9d9969f22729f
Diffstat (limited to 'source/use_case/img_class/src')
-rw-r--r-- | source/use_case/img_class/src/UseCaseHandler.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/source/use_case/img_class/src/UseCaseHandler.cc b/source/use_case/img_class/src/UseCaseHandler.cc index fafc6b9..1f1d78b 100644 --- a/source/use_case/img_class/src/UseCaseHandler.cc +++ b/source/use_case/img_class/src/UseCaseHandler.cc @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021 Arm Limited. All rights reserved. + * Copyright (c) 2021-2022 Arm Limited. All rights reserved. * SPDX-License-Identifier: Apache-2.0 * * Licensed under the Apache License, Version 2.0 (the "License"); @@ -19,6 +19,7 @@ #include "Classifier.hpp" #include "InputFiles.hpp" #include "MobileNetModel.hpp" +#include "ImageUtils.hpp" #include "UseCaseCommonUtils.hpp" #include "hal.h" #include "log_macros.h" @@ -136,7 +137,7 @@ namespace app { arm::app::DumpTensor(outputTensor); #endif /* VERIFY_TEST_OUTPUT */ - if (!image::PresentInferenceResult(platform, results)) { + if (!PresentInferenceResult(platform, results)) { return false; } |