| 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) | |