diff options
Diffstat (limited to 'source/use_case/vww')
-rw-r--r-- | source/use_case/vww/src/MainLoop.cc | 1 | ||||
-rw-r--r-- | source/use_case/vww/src/UseCaseHandler.cc | 4 | ||||
-rw-r--r-- | source/use_case/vww/src/VisualWakeWordModel.cc | 3 |
3 files changed, 5 insertions, 3 deletions
diff --git a/source/use_case/vww/src/MainLoop.cc b/source/use_case/vww/src/MainLoop.cc index b29238c..30e85bf 100644 --- a/source/use_case/vww/src/MainLoop.cc +++ b/source/use_case/vww/src/MainLoop.cc @@ -21,6 +21,7 @@ #include "VisualWakeWordModel.hpp" /* Model class for running inference. */ #include "UseCaseHandler.hpp" /* Handlers for different user options. */ #include "UseCaseCommonUtils.hpp" /* Utils functions. */ +#include "log_macros.h" using ViusalWakeWordClassifier = arm::app::Classifier; 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); diff --git a/source/use_case/vww/src/VisualWakeWordModel.cc b/source/use_case/vww/src/VisualWakeWordModel.cc index 3067c7a..59beccc 100644 --- a/source/use_case/vww/src/VisualWakeWordModel.cc +++ b/source/use_case/vww/src/VisualWakeWordModel.cc @@ -15,8 +15,7 @@ * limitations under the License. */ #include "VisualWakeWordModel.hpp" - -#include "hal.h" +#include "log_macros.h" const tflite::MicroOpResolver& arm::app::VisualWakeWordModel::GetOpResolver() { |