46 lines
No EOL
1.2 KiB
CMake
Executable file
46 lines
No EOL
1.2 KiB
CMake
Executable file
cmake_minimum_required(VERSION 3.5)
|
|
project(yolo11_demo)
|
|
|
|
set(CMAKE_CXX_STANDARD 17)
|
|
|
|
# Set NNSDK path
|
|
if(NOT DEFINED NNSDK_DIR)
|
|
set(NNSDK_DIR "${CMAKE_SOURCE_DIR}/../../../../../amlnn-toolkit/nn_runtime/nnsdk")
|
|
endif()
|
|
set(NNSDK_ROOT "${NNSDK_DIR}")
|
|
message(STATUS "NNSDK_ROOT: ${NNSDK_ROOT}")
|
|
|
|
include_directories(${NNSDK_ROOT}/include)
|
|
include_directories(${CMAKE_SOURCE_DIR}/../../../../common)
|
|
|
|
# Set dependency path
|
|
set(3RDPARTY_DIR "${CMAKE_SOURCE_DIR}/../../../../dependency")
|
|
|
|
if(CMAKE_SYSTEM_NAME STREQUAL "Android")
|
|
if (ANDROID_ABI STREQUAL "arm64-v8a")
|
|
link_directories(${NNSDK_ROOT}/android/arm64-v8a)
|
|
else()
|
|
link_directories(${NNSDK_ROOT}/android/armeabi-v7a)
|
|
endif()
|
|
# Android needs log
|
|
link_libraries(log)
|
|
elseif(CMAKE_SYSTEM_NAME STREQUAL "Linux")
|
|
link_directories(${NNSDK_ROOT}/linux/yocto/aarch64-poky-linux)
|
|
endif()
|
|
|
|
# Find OpenCV
|
|
message(STATUS "OpenCV_DIR: ${OpenCV_DIR}")
|
|
find_package(OpenCV REQUIRED)
|
|
include_directories(${OpenCV_INCLUDE_DIRS})
|
|
|
|
|
|
add_executable(yolo11_demo
|
|
main.cpp
|
|
postprocess.cpp
|
|
${CMAKE_SOURCE_DIR}/../../../../common/model_loader.cpp
|
|
)
|
|
|
|
target_link_libraries(yolo11_demo
|
|
${OpenCV_LIBS}
|
|
nnsdk
|
|
) |