MLECO-3171: Add flag to make VWW grayscale conversion optional
Signed-off-by: Richard Burton <richard.burton@arm.com>
Change-Id: I4ae40a60efc0c09f27b02f739769238e95bb112c
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 @@
/**
* @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 @@
private:
TfLiteTensor* m_inputTensor;
+ bool m_rgb2Gray;
};
/**