diff options
Diffstat (limited to 'source/use_case/vww/src/UseCaseHandler.cc')
-rw-r--r-- | source/use_case/vww/src/UseCaseHandler.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/source/use_case/vww/src/UseCaseHandler.cc b/source/use_case/vww/src/UseCaseHandler.cc index e4dc479..01011e2 100644 --- a/source/use_case/vww/src/UseCaseHandler.cc +++ b/source/use_case/vww/src/UseCaseHandler.cc @@ -20,6 +20,7 @@ #include "InputFiles.hpp" #include "UseCaseCommonUtils.hpp" #include "hal.h" +#include "log_macros.h" #include <algorithm> @@ -133,7 +134,8 @@ namespace app { auto& classifier = ctx.Get<Classifier&>("classifier"); classifier.GetClassificationResults(outputTensor, results, - ctx.Get<std::vector <std::string>&>("labels"), 1); + ctx.Get<std::vector <std::string>&>("labels"), 1, + false); /* Add results to context for access outside handler. */ ctx.Set<std::vector<ClassificationResult>>("results", results); |