| cmake_minimum_required(VERSION 3.5) |
|
|
| project(Yolov8CPPInference VERSION 0.1) |
|
|
| set(CMAKE_INCLUDE_CURRENT_DIR ON) |
|
|
| # CUDA |
| set(CUDA_TOOLKIT_ROOT_DIR "/usr/local/cuda") |
| find_package(CUDA 11 REQUIRED) |
|
|
| set(CMAKE_CUDA_STANDARD 11) |
| set(CMAKE_CUDA_STANDARD_REQUIRED ON) |
| # !CUDA |
|
|
| # OpenCV |
| find_package(OpenCV REQUIRED) |
| include_directories(${OpenCV_INCLUDE_DIRS}) |
| # !OpenCV |
|
|
| set(PROJECT_SOURCES |
| main.cpp |
|
|
| inference.h |
| inference.cpp |
| ) |
|
|
| add_executable(Yolov8CPPInference ${PROJECT_SOURCES}) |
| target_link_libraries(Yolov8CPPInference ${OpenCV_LIBS}) |
|
|
