diff options
Diffstat (limited to 'source/use_case/img_class/usecase.cmake')
-rw-r--r-- | source/use_case/img_class/usecase.cmake | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/source/use_case/img_class/usecase.cmake b/source/use_case/img_class/usecase.cmake index 4c25989..f18522c 100644 --- a/source/use_case/img_class/usecase.cmake +++ b/source/use_case/img_class/usecase.cmake @@ -1,5 +1,5 @@ #---------------------------------------------------------------------------- -# SPDX-FileCopyrightText: Copyright 2021 Arm Limited and/or its affiliates <open-source-office@arm.com> +# SPDX-FileCopyrightText: Copyright 2021, 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"); @@ -28,10 +28,12 @@ USER_OPTION(${use_case}_LABELS_TXT_FILE "Labels' txt file for the chosen model" FILEPATH) # Generate input files -generate_images_code("${${use_case}_FILE_PATH}" - ${SRC_GEN_DIR} - ${INC_GEN_DIR} - "${${use_case}_IMAGE_SIZE}") +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 labels file set(${use_case}_LABELS_CPP_FILE Labels) |