diff options
Diffstat (limited to 'source/use_case/object_detection/src')
-rw-r--r-- | source/use_case/object_detection/src/MainLoop.cc | 2 | ||||
-rw-r--r-- | source/use_case/object_detection/src/UseCaseHandler.cc | 6 | ||||
-rw-r--r-- | source/use_case/object_detection/src/YoloFastestModel.cc | 2 |
3 files changed, 6 insertions, 4 deletions
diff --git a/source/use_case/object_detection/src/MainLoop.cc b/source/use_case/object_detection/src/MainLoop.cc index d8fc7f5..4bec357 100644 --- a/source/use_case/object_detection/src/MainLoop.cc +++ b/source/use_case/object_detection/src/MainLoop.cc @@ -20,7 +20,7 @@ #include "UseCaseHandler.hpp" /* Handlers for different user options. */ #include "UseCaseCommonUtils.hpp" /* Utils functions. */ #include "DetectorPostProcessing.hpp" /* Post-processing class. */ - +#include "log_macros.h" static void DisplayDetectionMenu() { diff --git a/source/use_case/object_detection/src/UseCaseHandler.cc b/source/use_case/object_detection/src/UseCaseHandler.cc index ce3ef06..620ce6c 100644 --- a/source/use_case/object_detection/src/UseCaseHandler.cc +++ b/source/use_case/object_detection/src/UseCaseHandler.cc @@ -20,8 +20,9 @@ #include "UseCaseCommonUtils.hpp" #include "DetectorPostProcessing.hpp" #include "hal.h" +#include "log_macros.h" -#include <inttypes.h> +#include <cinttypes> namespace arm { namespace app { @@ -151,7 +152,8 @@ namespace app { dataPsnImgStartX, dataPsnImgStartY, dataPsnImgDownscaleFactor); #if VERIFY_TEST_OUTPUT - arm::app::DumpTensor(outputTensor); + arm::app::DumpTensor(modelOutput0); + arm::app::DumpTensor(modelOutput1); #endif /* VERIFY_TEST_OUTPUT */ if (!PresentInferenceResult(platform, results)) { diff --git a/source/use_case/object_detection/src/YoloFastestModel.cc b/source/use_case/object_detection/src/YoloFastestModel.cc index a8afd59..b1fd776 100644 --- a/source/use_case/object_detection/src/YoloFastestModel.cc +++ b/source/use_case/object_detection/src/YoloFastestModel.cc @@ -16,7 +16,7 @@ */ #include "YoloFastestModel.hpp" -#include "hal.h" +#include "log_macros.h" const tflite::MicroOpResolver& arm::app::YoloFastestModel::GetOpResolver() { |