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/src | |
| 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/src')
| -rw-r--r-- | source/use_case/kws/src/KwsProcessing.cc | 5 | ||||
| -rw-r--r-- | source/use_case/kws/src/UseCaseHandler.cc | 4 |
2 files changed, 4 insertions, 5 deletions
diff --git a/source/use_case/kws/src/KwsProcessing.cc b/source/use_case/kws/src/KwsProcessing.cc index b6b230c..14f9fce 100644 --- a/source/use_case/kws/src/KwsProcessing.cc +++ b/source/use_case/kws/src/KwsProcessing.cc @@ -197,11 +197,10 @@ namespace app { KWSPostProcess::KWSPostProcess(Classifier& classifier, Model* model, const std::vector<std::string>& labels, - std::vector<ClassificationResult>& results, float scoreThreshold) + std::vector<ClassificationResult>& results) :m_kwsClassifier{classifier}, m_labels{labels}, - m_results{results}, - m_scoreThreshold{scoreThreshold} + m_results{results} { if (!model->IsInited()) { printf_err("Model is not initialised!.\n"); diff --git a/source/use_case/kws/src/UseCaseHandler.cc b/source/use_case/kws/src/UseCaseHandler.cc index 350d34b..e73a2c3 100644 --- a/source/use_case/kws/src/UseCaseHandler.cc +++ b/source/use_case/kws/src/UseCaseHandler.cc @@ -93,7 +93,7 @@ namespace app { std::vector<ClassificationResult> singleInfResult; KWSPostProcess postprocess = KWSPostProcess(ctx.Get<KwsClassifier &>("classifier"), &model, ctx.Get<std::vector<std::string>&>("labels"), - singleInfResult, scoreThreshold); + singleInfResult); UseCaseRunner runner = UseCaseRunner(&preprocess, &postprocess, &model); @@ -146,7 +146,7 @@ namespace app { /* Add results from this window to our final results vector. */ finalResults.emplace_back(kws::KwsResult(singleInfResult, audioDataSlider.Index() * secondsPerSample * preprocess.m_audioDataStride, - audioDataSlider.Index(), postprocess.m_scoreThreshold)); + audioDataSlider.Index(), scoreThreshold)); #if VERIFY_TEST_OUTPUT TfLiteTensor* outputTensor = model.GetOutputTensor(0); |