summaryrefslogtreecommitdiff
path: root/tests/use_case/img_class
diff options
context:
space:
mode:
authorRichard Burton <richard.burton@arm.com>2022-10-05 11:00:37 +0100
committerRichard Burton <richard.burton@arm.com>2022-10-06 14:08:13 +0100
commitec5e99be3ae6dd0d3811950f155b01e144431452 (patch)
treea5d6c4dd9267db2465063b8d0e1a5cb6d19dac8d /tests/use_case/img_class
parent890b2b89cacc6f2291596a001d555d374c8c9edd (diff)
downloadml-embedded-evaluation-kit-ec5e99be3ae6dd0d3811950f155b01e144431452.tar.gz
MLECO-3164: Additional refactoring of KWS API
Part 1 * Add KwsClassifier * KwsPostProcess can now be told to average results * Averaging is handlded by KwsClassifier * Current sliding window index is now an argument of DoPreProcess Change-Id: I07626da595ad1cbd982e8366f0d1bb56d1040459
Diffstat (limited to 'tests/use_case/img_class')
-rw-r--r--tests/use_case/img_class/ImgClassificationUCTest.cc3
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/use_case/img_class/ImgClassificationUCTest.cc b/tests/use_case/img_class/ImgClassificationUCTest.cc
index 1685e5f..78700dc 100644
--- a/tests/use_case/img_class/ImgClassificationUCTest.cc
+++ b/tests/use_case/img_class/ImgClassificationUCTest.cc
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2021 Arm Limited. All rights reserved.
+ * Copyright (c) 2021-2022 Arm Limited. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*
* Licensed under the Apache License, Version 2.0 (the "License");
@@ -21,6 +21,7 @@
#include "MobileNetModel.hpp"
#include "UseCaseHandler.hpp"
#include "UseCaseCommonUtils.hpp"
+#include "BufAttributes.hpp"
#include <catch.hpp>