diff options
author | Richard Burton <richard.burton@arm.com> | 2022-04-22 09:08:21 +0100 |
---|---|---|
committer | Richard Burton <richard.burton@arm.com> | 2022-04-22 09:08:21 +0100 |
commit | c291144b7f08c21d08cdaf79cc64dc420ca70070 (patch) | |
tree | 1b91c38f7dd479a0c13772a1e1da52079d06237c /source/use_case/kws/include/KwsProcessing.hpp | |
parent | b1904b11d15da48c7ead4e6bb85c3e671956ab03 (diff) | |
download | ml-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/KwsProcessing.hpp')
-rw-r--r-- | source/use_case/kws/include/KwsProcessing.hpp | 23 |
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; }; |