| cmake_minimum_required(VERSION 2.6) |
|
|
| project(mcnet) |
|
|
| add_definitions(-std=c++11) |
|
|
| option(CUDA_USE_STATIC_CUDA_RUNTIME OFF) |
| set(CMAKE_CXX_STANDARD 11) |
| set(CMAKE_BUILD_TYPE Release) |
|
|
|
|
| find_package(ZED 3 REQUIRED) |
| find_package(CUDA ${ZED_CUDA_VERSION} EXACT REQUIRED) |
|
|
| include_directories(${PROJECT_SOURCE_DIR}/include) |
|
|
| # cuda |
| include_directories(/usr/local/cuda-10.2/include) |
| link_directories(/usr/local/cuda-10.2/lib64) |
| # tensorrt |
| include_directories(/usr/include/aarch64-linux-gnu/) |
| link_directories(/usr/lib/aarch64-linux-gnu/) |
| # zed |
| include_directories(/usr/local/zed/include) |
| link_directories(/usr/local/zed/lib) |
|
|
| set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Ofast -Wfatal-errors -D_MWAITXINTRIN_H_INCLUDED") |
|
|
| set(ZED_LIBS ${ZED_LIBRARIES} ${CUDA_CUDA_LIBRARY} ${CUDA_CUDART_LIBRARY}) |
|
|
| coda_add_library(myplugins SHARED ${PROJECT_SOURCE_DIR}/yololayer.cu) |
| target_link_libraries(myplugins nvinfer cudart) |
|
|
| find_package(OpenCV REQUIRED) |
| include_directories(${OpenCV_INCLUDE_DIRS}) |
|
|
| add_executable(mcnet ${PROJECT_SOURCE_DIR}/main.cpp) |
| target_link_libraries(mcnet nvinfer) |
| target_link_libraries(mcnet ${ZED_LIBS}) |
| target_link_libraries(mcnet cudart) |
| target_link_libraries(mcnet myplugins) |
| target_link_libraries(mcnet ${OpenCV_LIBS}) |
|
|
| add_definitions(-O3 -pthread) |
|
|
|
|
