diff options
| author | Isabella Gottardi <isabella.gottardi@arm.com> | 2022-01-27 16:39:37 +0000 |
|---|---|---|
| committer | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2022-02-08 16:32:28 +0000 |
| commit | 3107aa2152de9be8317e62da1d0327bcad6552e2 (patch) | |
| tree | 2ba12a5dd39f28ae1b646e132fbe575c6a442ee9 /source/use_case/vww/include | |
| parent | 5cdfa9b834dc5a94c70f9f2b1f5c849dc5439e85 (diff) | |
| download | ml-embedded-evaluation-kit-3107aa2152de9be8317e62da1d0327bcad6552e2.tar.gz | |
MLECO-2873: Object detection usecase follow-up
Change-Id: Ic14e93a50fb7b3f3cfd9497bac1280794cc0fc15
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Diffstat (limited to 'source/use_case/vww/include')
| -rw-r--r-- | source/use_case/vww/include/VisualWakeWordModel.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/source/use_case/vww/include/VisualWakeWordModel.hpp b/source/use_case/vww/include/VisualWakeWordModel.hpp index ee3a7bf..1ed9202 100644 --- a/source/use_case/vww/include/VisualWakeWordModel.hpp +++ b/source/use_case/vww/include/VisualWakeWordModel.hpp @@ -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"); @@ -24,6 +24,12 @@ namespace app { class VisualWakeWordModel : public Model { + public: + /* Indices for the expected model - based on input tensor shape */ + static constexpr uint32_t ms_inputRowsIdx = 1; + static constexpr uint32_t ms_inputColsIdx = 2; + static constexpr uint32_t ms_inputChannelsIdx = 3; + protected: /** @brief Gets the reference to op resolver interface class. */ const tflite::MicroOpResolver& GetOpResolver() override; |