1cmake_minimum_required(VERSION 3.4.1) 2 3project(rknn_yolov5_demo) 4 5set(CMAKE_CXX_STANDARD 11) 6set(CMAKE_CXX_STANDARD_REQUIRED ON) 7 8# skip 3rd-party lib dependencies 9set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Wl,--allow-shlib-undefined") 10 11# install target and libraries 12set(CMAKE_INSTALL_PREFIX ${CMAKE_SOURCE_DIR}/install/rknn_yolov5_demo_${CMAKE_SYSTEM_NAME}) 13 14set(CMAKE_SKIP_INSTALL_RPATH FALSE) 15set(CMAKE_BUILD_WITH_INSTALL_RPATH TRUE) 16set(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_PREFIX}/lib") 17 18if (CMAKE_C_COMPILER MATCHES "aarch64") 19 set(LIB_ARCH aarch64) 20else() 21 set(LIB_ARCH armhf) 22endif() 23 24include_directories(${CMAKE_SOURCE_DIR}) 25 26# rknn api 27if(TARGET_SOC STREQUAL "rk356x") 28 set(RKNN_API_PATH ${CMAKE_SOURCE_DIR}/../../runtime/RK356X/${CMAKE_SYSTEM_NAME}/librknn_api) 29elseif(TARGET_SOC STREQUAL "rk3588") 30 set(RKNN_API_PATH ${CMAKE_SOURCE_DIR}/../../runtime/RK3588/${CMAKE_SYSTEM_NAME}/librknn_api) 31else() 32 message(FATAL_ERROR "TARGET_SOC is not set, ref value: rk356x or rk3588 or rv110x") 33endif() 34 35if (CMAKE_SYSTEM_NAME STREQUAL "Android") 36 set(RKNN_RT_LIB ${RKNN_API_PATH}/${CMAKE_ANDROID_ARCH_ABI}/librknnrt.so) 37else() 38 set(RKNN_RT_LIB ${RKNN_API_PATH}/${LIB_ARCH}/librknnrt.so) 39endif() 40include_directories(${RKNN_API_PATH}/include) 41include_directories(${CMAKE_SOURCE_DIR}/../3rdparty) 42 43# opencv 44if (CMAKE_SYSTEM_NAME STREQUAL "Android") 45 set(OpenCV_DIR ${CMAKE_SOURCE_DIR}/../3rdparty/opencv/OpenCV-android-sdk/sdk/native/jni/abi-${CMAKE_ANDROID_ARCH_ABI}) 46else() 47 if(LIB_ARCH STREQUAL "armhf") 48 set(OpenCV_DIR ${CMAKE_SOURCE_DIR}/../3rdparty/opencv/opencv-linux-armhf/share/OpenCV) 49 else() 50 set(OpenCV_DIR ${CMAKE_SOURCE_DIR}/../3rdparty/opencv/opencv-linux-aarch64/share/OpenCV) 51 endif() 52endif() 53find_package(OpenCV REQUIRED) 54 55#rga 56if(TARGET_SOC STREQUAL "rk356x") 57 set(RGA_PATH ${CMAKE_SOURCE_DIR}/../3rdparty/rga/RK356X) 58elseif(TARGET_SOC STREQUAL "rk3588") 59 set(RGA_PATH ${CMAKE_SOURCE_DIR}/../3rdparty/rga/RK3588) 60else() 61 message(FATAL_ERROR "TARGET_SOC is not set, ref value: rk356x or rk3588") 62endif() 63if (CMAKE_SYSTEM_NAME STREQUAL "Android") 64 set(RGA_LIB ${RGA_PATH}/lib/Android/${CMAKE_ANDROID_ARCH_ABI}/librga.so) 65else() 66 set(RGA_LIB ${RGA_PATH}/lib/Linux//${LIB_ARCH}/librga.so) 67endif() 68include_directories( ${RGA_PATH}/include) 69 70# mpp 71set(MPP_PATH ${CMAKE_CURRENT_SOURCE_DIR}/../3rdparty/mpp) 72if (CMAKE_SYSTEM_NAME STREQUAL "Linux") 73 set(MPP_LIBS ${MPP_PATH}/${CMAKE_SYSTEM_NAME}/${LIB_ARCH}/librockchip_mpp.so) 74elseif (CMAKE_SYSTEM_NAME STREQUAL "Android") 75 set(MPP_LIBS ${MPP_PATH}/${CMAKE_SYSTEM_NAME}/${CMAKE_ANDROID_ARCH_ABI}/libmpp.so) 76endif() 77include_directories(${MPP_PATH}/include) 78 79# zlmediakit 80set(ZLMEDIAKIT_PATH ${CMAKE_SOURCE_DIR}/../3rdparty/zlmediakit) 81if (CMAKE_SYSTEM_NAME STREQUAL "Linux") 82include_directories(${ZLMEDIAKIT_PATH}/include) 83set(ZLMEDIAKIT_LIBS ${ZLMEDIAKIT_PATH}/${LIB_ARCH}/libmk_api.so) 84endif() 85 86if (ZLMEDIAKIT_LIBS) 87 add_definitions(-DBUILD_VIDEO_RTSP) 88endif() 89 90set(CMAKE_INSTALL_RPATH "lib") 91 92# rknn_yolov5_demo 93include_directories( ${CMAKE_SOURCE_DIR}/include) 94 95add_executable(rknn_yolov5_demo 96 src/main.cc 97 src/postprocess.cc 98) 99 100target_link_libraries(rknn_yolov5_demo 101 ${RKNN_RT_LIB} 102 ${RGA_LIB} 103 ${OpenCV_LIBS} 104) 105 106if (MPP_LIBS) 107add_executable(rknn_yolov5_video_demo 108 src/main_video.cc 109 src/postprocess.cc 110 utils/mpp_decoder.cpp 111 utils/mpp_encoder.cpp 112 utils/drawing.cpp 113) 114target_link_libraries(rknn_yolov5_video_demo 115 ${RKNN_RT_LIB} 116 ${RGA_LIB} 117 ${OpenCV_LIBS} 118 ${MPP_LIBS} 119 ${ZLMEDIAKIT_LIBS} 120) 121endif() 122 123 124# install target and libraries 125set(CMAKE_INSTALL_PREFIX ${CMAKE_SOURCE_DIR}/install/rknn_yolov5_demo_${CMAKE_SYSTEM_NAME}) 126install(TARGETS rknn_yolov5_demo DESTINATION ./) 127 128install(PROGRAMS ${RKNN_RT_LIB} DESTINATION lib) 129install(PROGRAMS ${RGA_LIB} DESTINATION lib) 130install(DIRECTORY model DESTINATION ./) 131 132if (MPP_LIBS) 133install(TARGETS rknn_yolov5_video_demo DESTINATION ./) 134install(PROGRAMS ${MPP_LIBS} DESTINATION lib) 135endif() 136 137if (ZLMEDIAKIT_LIBS) 138install(PROGRAMS ${ZLMEDIAKIT_LIBS} DESTINATION lib) 139endif()