summaryrefslogtreecommitdiff
path: root/source/use_case/kws/include
diff options
context:
space:
mode:
authorRichard Burton <richard.burton@arm.com>2022-04-22 09:08:21 +0100
committerRichard Burton <richard.burton@arm.com>2022-04-22 09:08:21 +0100
commitc291144b7f08c21d08cdaf79cc64dc420ca70070 (patch)
tree1b91c38f7dd479a0c13772a1e1da52079d06237c /source/use_case/kws/include
parentb1904b11d15da48c7ead4e6bb85c3e671956ab03 (diff)
downloadml-embedded-evaluation-kit-c291144b7f08c21d08cdaf79cc64dc420ca70070.tar.gz
MLECO-3077: Add ASR use case API
* Minor adjustments to doc strings in KWS * Remove unused score threshold in KWS Signed-off-by: Richard Burton <richard.burton@arm.com> Change-Id: Ie1c5bf6f7bdbebb853b6a10cb7ba1c4a1d9a76c9
Diffstat (limited to 'source/use_case/kws/include')
-rw-r--r--source/use_case/kws/include/KwsProcessing.hpp23
1 files changed, 10 insertions, 13 deletions
diff --git a/source/use_case/kws/include/KwsProcessing.hpp b/source/use_case/kws/include/KwsProcessing.hpp
index abf20ab..ddf38c1 100644
--- a/source/use_case/kws/include/KwsProcessing.hpp
+++ b/source/use_case/kws/include/KwsProcessing.hpp
@@ -38,7 +38,7 @@ namespace app {
public:
/**
* @brief Constructor
- * @param[in] model Pointer to the the KWS Model object.
+ * @param[in] model Pointer to the KWS Model object.
* @param[in] numFeatures How many MFCC features to use.
* @param[in] mfccFrameLength Number of audio samples used to calculate one set of MFCC values when
* sliding a window through the audio sample.
@@ -107,24 +107,21 @@ namespace app {
std::vector<ClassificationResult>& m_results;
public:
- const float m_scoreThreshold;
/**
- * @brief Constructor
- * @param[in] classifier Classifier object used to get top N results from classification.
- * @param[in] model Pointer to the the Image classification Model object.
- * @param[in] labels Vector of string labels to identify each output of the model.
- * @param[in] results Vector of classification results to store decoded outputs.
- * @param[in] scoreThreshold Predicted model score must be larger than this value to be accepted.
+ * @brief Constructor
+ * @param[in] classifier Classifier object used to get top N results from classification.
+ * @param[in] model Pointer to the KWS Model object.
+ * @param[in] labels Vector of string labels to identify each output of the model.
+ * @param[in/out] results Vector of classification results to store decoded outputs.
**/
KWSPostProcess(Classifier& classifier, Model* model,
const std::vector<std::string>& labels,
- std::vector<ClassificationResult>& results,
- float scoreThreshold);
+ std::vector<ClassificationResult>& results);
/**
- * @brief Should perform post-processing of the result of inference then populate
- * populate KWS result data for any later use.
- * @return true if successful, false otherwise.
+ * @brief Should perform post-processing of the result of inference then
+ * populate KWS result data for any later use.
+ * @return true if successful, false otherwise.
**/
bool DoPostProcess() override;
};