diff options
Diffstat (limited to 'source/use_case/kws/src/UseCaseHandler.cc')
-rw-r--r-- | source/use_case/kws/src/UseCaseHandler.cc | 4 |
1 files changed, 2 insertions, 2 deletions
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); |