diff options
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; }; /** |