summaryrefslogtreecommitdiff
path: root/source/use_case/object_detection/src
diff options
context:
space:
mode:
Diffstat (limited to 'source/use_case/object_detection/src')
-rw-r--r--source/use_case/object_detection/src/MainLoop.cc2
-rw-r--r--source/use_case/object_detection/src/UseCaseHandler.cc6
-rw-r--r--source/use_case/object_detection/src/YoloFastestModel.cc2
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()
{