diff options
| author | Richard Burton <richard.burton@arm.com> | 2022-05-05 17:23:43 +0100 |
|---|---|---|
| committer | Richard Burton <richard.burton@arm.com> | 2022-05-05 17:23:43 +0100 |
| commit | fcca863bafd5f33522bc14c23dde4540e264ec94 (patch) | |
| tree | 8d15ee9000ef9fd0fa21314650900723cefc8b36 /source/use_case/vww/include/VisualWakeWordProcessing.hpp | |
| parent | 4e002791bc6781b549c6951cfe44f918289d7e82 (diff) | |
| download | ml-embedded-evaluation-kit-fcca863bafd5f33522bc14c23dde4540e264ec94.tar.gz | |
MLECO-3171: Add flag to make VWW grayscale conversion optional
Signed-off-by: Richard Burton <richard.burton@arm.com>
Change-Id: I4ae40a60efc0c09f27b02f739769238e95bb112c
Diffstat (limited to 'source/use_case/vww/include/VisualWakeWordProcessing.hpp')
| -rw-r--r-- | source/use_case/vww/include/VisualWakeWordProcessing.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/source/use_case/vww/include/VisualWakeWordProcessing.hpp b/source/use_case/vww/include/VisualWakeWordProcessing.hpp index bef161f..f9f9d72 100644 --- a/source/use_case/vww/include/VisualWakeWordProcessing.hpp +++ b/source/use_case/vww/include/VisualWakeWordProcessing.hpp @@ -35,8 +35,9 @@ namespace app { /** * @brief Constructor * @param[in] inputTensor Pointer to the TFLite Micro input Tensor. + * @param[in] rgb2Gray Convert image from 3 channel RGB to 1 channel grayscale. **/ - explicit VisualWakeWordPreProcess(TfLiteTensor* inputTensor); + explicit VisualWakeWordPreProcess(TfLiteTensor* inputTensor, bool rgb2Gray=true); /** * @brief Should perform pre-processing of 'raw' input image data and load it into @@ -49,6 +50,7 @@ namespace app { private: TfLiteTensor* m_inputTensor; + bool m_rgb2Gray; }; /** |