diff options
Diffstat (limited to 'source/use_case/object_detection/usecase.cmake')
-rw-r--r-- | source/use_case/object_detection/usecase.cmake | 24 |
1 files changed, 19 insertions, 5 deletions
diff --git a/source/use_case/object_detection/usecase.cmake b/source/use_case/object_detection/usecase.cmake index a36d3e0..cb0236b 100644 --- a/source/use_case/object_detection/usecase.cmake +++ b/source/use_case/object_detection/usecase.cmake @@ -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"); @@ -36,10 +36,24 @@ USER_OPTION(${use_case}_CHANNELS_IMAGE_DISPLAYED "Channels' image displayed on t BOOL) # Generate input files -generate_images_code("${${use_case}_FILE_PATH}" - ${SRC_GEN_DIR} - ${INC_GEN_DIR} - "${${use_case}_IMAGE_SIZE}") +if (VSI_IMAGE_INPUT) + set(${use_case}_COMPILE_DEFS "VSI_IMAGE_INPUT") + + generate_images_code( + INPUT_DIR "${${use_case}_FILE_PATH}" + SRC_OUT ${SRC_GEN_DIR} + HDR_OUT ${INC_GEN_DIR} + IMG_SIZE "${${use_case}_IMAGE_SIZE}" + GENERATE_FILE_PATHS + ) +else () + generate_images_code( + INPUT_DIR "${${use_case}_FILE_PATH}" + SRC_OUT ${SRC_GEN_DIR} + HDR_OUT ${INC_GEN_DIR} + IMG_SIZE "${${use_case}_IMAGE_SIZE}" + ) +endif () USER_OPTION(${use_case}_ACTIVATION_BUF_SZ "Activation buffer size for the chosen model" 0x00082000 |