diff options
Diffstat (limited to 'source/use_case/object_detection/src/MainLoop.cc')
-rw-r--r-- | source/use_case/object_detection/src/MainLoop.cc | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/source/use_case/object_detection/src/MainLoop.cc b/source/use_case/object_detection/src/MainLoop.cc index 4735bcb..fe21de6 100644 --- a/source/use_case/object_detection/src/MainLoop.cc +++ b/source/use_case/object_detection/src/MainLoop.cc @@ -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"); @@ -41,7 +41,12 @@ static void DisplayDetectionMenu() printf(" %u. Run detection ifm at chosen index\n", common::MENU_OPT_RUN_INF_CHOSEN); printf(" %u. Run detection on all ifm\n", common::MENU_OPT_RUN_INF_ALL); printf(" %u. Show NN model info\n", common::MENU_OPT_SHOW_MODEL_INFO); - printf(" %u. List ifm\n\n", common::MENU_OPT_LIST_IFM); + printf(" %u. List ifm\n", common::MENU_OPT_LIST_IFM); +#if VSI_ENABLED + fflush(stdout); + printf(" %u. Run detection using VSI as input\n", common::MENU_OPT_RUN_INF_VSI); +#endif + printf("\n"); printf(" Choice: "); fflush(stdout); } @@ -99,6 +104,11 @@ void main_loop() case common::MENU_OPT_LIST_IFM: executionSuccessful = ListFilesHandler(caseContext); break; +#if VSI_ENABLED + case common::MENU_OPT_RUN_INF_VSI: + executionSuccessful = ObjectDetectionHandlerVsi(caseContext); + break; +#endif default: printf("Incorrect choice, try again."); break; |