diff options
| author | Richard Burton <richard.burton@arm.com> | 2023-12-06 17:13:10 +0000 |
|---|---|---|
| committer | alex.tawse <alex.tawse@arm.com> | 2024-01-23 17:26:45 +0000 |
| commit | cefc7e1cacdd3028b46325b3a1f6c15416914b2f (patch) | |
| tree | d0353f7775d18a207d8a9fd352a3b716ea7b9773 /source/use_case/object_detection/include/UseCaseHandler.hpp | |
| parent | 001a8ff315dc766a206b05a3e00e831e3f972b0d (diff) | |
| download | ml-embedded-evaluation-kit-cefc7e1cacdd3028b46325b3a1f6c15416914b2f.tar.gz | |
MLECO-4503: Adding video VSI for object detectionexperimental/vsi
* Added Board support - Arm Corstone 300 and 310
* Added Python Scripts for Video VSI
* Added source files for Video VSI
* Add new usecase handler for OD use case
* Bumped resampy version to resolve issue with slowdown
Signed-off-by: Idriss Chaouch <idriss.chaouch@arm.com>
Signed-off-by: Richard Burton <richard.burton@arm.com>
Change-Id: Ie59ae955d4d85f672a49c63733052624542aec85
Diffstat (limited to 'source/use_case/object_detection/include/UseCaseHandler.hpp')
| -rw-r--r-- | source/use_case/object_detection/include/UseCaseHandler.hpp | 30 |
1 files changed, 29 insertions, 1 deletions
diff --git a/source/use_case/object_detection/include/UseCaseHandler.hpp b/source/use_case/object_detection/include/UseCaseHandler.hpp index 9fe716f..43fb6bf 100644 --- a/source/use_case/object_detection/include/UseCaseHandler.hpp +++ b/source/use_case/object_detection/include/UseCaseHandler.hpp @@ -1,5 +1,5 @@ /* - * SPDX-FileCopyrightText: Copyright 2022 Arm Limited and/or its affiliates <open-source-office@arm.com> + * SPDX-FileCopyrightText: Copyright 2022, 2024 Arm Limited and/or its affiliates <open-source-office@arm.com> * SPDX-License-Identifier: Apache-2.0 * * Licensed under the Apache License, Version 2.0 (the "License"); @@ -17,11 +17,39 @@ #ifndef OBJ_DET_HANDLER_HPP #define OBJ_DET_HANDLER_HPP +#if VSI_ENABLED +#include "DetectionResult.hpp" +#include <vector> +#endif + +#include <cstdint> #include "AppContext.hpp" namespace arm { namespace app { +#if VSI_ENABLED + typedef object_detection::DetectionResult OdResults; + + /** + * @brief Draw boxe(s) detected by the model. + * @param[in] image Pointer to the image. + * @param[in] imageWidth Image width. + * @param[in] imageHeight Image height. + * @param[in] results A vector of detection results. + **/ + void DrawDetectionBoxesVsi(uint8_t* image, + const uint32_t imageWidth, + const uint32_t imageHeight, + const std::vector<OdResults>& results); + /** + * @brief Handles the inference event when using VSI. + * @param[in] ctx Pointer to the application context. + * @return true or false based on execution success. + **/ + bool ObjectDetectionHandlerVsi(ApplicationContext& ctx); +#endif + /** * @brief Handles the inference event. * @param[in] ctx Pointer to the application context. |