diff options
| author | George Gekov <george.gekov@arm.com> | 2021-08-16 11:32:10 +0100 |
|---|---|---|
| committer | Jim Flynn <jim.flynn@arm.com> | 2022-02-05 19:49:06 +0000 |
| commit | 23c26277086c78704a17f0dae86da947816320c0 (patch) | |
| tree | 88b02fd1fae3130256d059251788a7ef68d2831f /samples/ObjectDetection/cmake/unit_tests.cmake | |
| parent | 922b912fd2d462bac0809bac5669310ad1506310 (diff) | |
| download | armnn-23c26277086c78704a17f0dae86da947816320c0.tar.gz | |
MLECO-2079 Adding the C++ KWS example
Signed-off-by: Eanna O Cathain <eanna.ocathain@arm.com>
Change-Id: I81899bbfaada32f478c2e2fc6441eabb94d8d0fc
Diffstat (limited to 'samples/ObjectDetection/cmake/unit_tests.cmake')
| -rw-r--r-- | samples/ObjectDetection/cmake/unit_tests.cmake | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/samples/ObjectDetection/cmake/unit_tests.cmake b/samples/ObjectDetection/cmake/unit_tests.cmake index 1a8c466d6b..dd3de708ba 100644 --- a/samples/ObjectDetection/cmake/unit_tests.cmake +++ b/samples/ObjectDetection/cmake/unit_tests.cmake @@ -2,6 +2,7 @@ # SPDX-License-Identifier: MIT set(TEST_RESOURCES_DIR ${CMAKE_SOURCE_DIR}/test/resources) +file(MAKE_DIRECTORY ${TEST_RESOURCES_DIR}) add_definitions (-DTEST_RESOURCE_DIR="${TEST_RESOURCES_DIR}") set(TEST_TARGET_NAME "${CMAKE_PROJECT_NAME}-tests") @@ -9,16 +10,6 @@ file(GLOB TEST_SOURCES "test/*") include(../common/cmake/find_catch.cmake) -file(DOWNLOAD "https://storage.googleapis.com/download.tensorflow.org/models/tflite/coco_ssd_mobilenet_v1_1.0_quant_2018_06_29.zip" - ${CMAKE_CURRENT_SOURCE_DIR}/test/resources/models.zip SHOW_PROGRESS) - -# Extract -execute_process( - COMMAND ${CMAKE_COMMAND} -E tar xf models.zip - WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/test/resources/ - RESULT_VARIABLE return_code -) - ExternalProject_Add(basketball-image URL https://raw.githubusercontent.com/opencv/opencv/4.0.0/samples/data/basketball1.png DOWNLOAD_NO_EXTRACT 1 @@ -43,7 +34,15 @@ ExternalProject_Add(vtest INSTALL_COMMAND "" ) -add_executable("${TEST_TARGET_NAME}" ${SOURCES} ${TEST_SOURCES} ${COMMON_SOURCES}) +ExternalProject_Add(ssd_mobile + URL https://github.com/ARM-software/ML-zoo/raw/master/models/object_detection/ssd_mobilenet_v1/tflite_uint8/ssd_mobilenet_v1.tflite + DOWNLOAD_NO_EXTRACT 1 + CONFIGURE_COMMAND "" + BUILD_COMMAND ${CMAKE_COMMAND} -E copy <DOWNLOAD_DIR>/ssd_mobilenet_v1.tflite ${CMAKE_CURRENT_SOURCE_DIR}/test/resources + INSTALL_COMMAND "" + ) + +add_executable("${TEST_TARGET_NAME}" ${SOURCES} ${TEST_SOURCES} ${CVUTILS_SOURCES} ${UTILS_SOURCES}) add_dependencies( "${TEST_TARGET_NAME}" |