CMakeLists.txt 823 B

123456789101112131415161718192021222324252627
  1. CMAKE_MINIMUM_REQUIRED(VERSION 3.0.0)
  2. project(YOLOv8)
  3. SET (OpenCV_DIR path/to/opencv/build) #opencv root
  4. SET (ONNXRUNTIME_DIR path/to/onnxruntime)
  5. FIND_PACKAGE(OpenCV REQUIRED)
  6. #include_directories("")
  7. ADD_EXECUTABLE(YOLOv8 yolov8.h yolov8_onnx.h yolov8_seg.h yolov8_seg_onnx.h yolov8_utils.h
  8. main.cpp yolov8.cpp yolov8_onnx.cpp yolov8_seg.cpp yolov8_seg_onnx.cpp yolov8_utils.cpp)
  9. SET(CMAKE_CXX_STANDARD 14)
  10. SET(CMAKE_CXX_STANDARD_REQUIRED ON)
  11. TARGET_INCLUDE_DIRECTORIES(YOLOv8 PRIVATE "${ONNXRUNTIME_DIR}/include")
  12. TARGET_COMPILE_FEATURES(YOLOv8 PRIVATE cxx_std_14)
  13. TARGET_LINK_LIBRARIES(YOLOv8 ${OpenCV_LIBS})
  14. if (WIN32)
  15. TARGET_LINK_LIBRARIES(YOLOv8 "${ONNXRUNTIME_DIR}/lib/onnxruntime.lib")
  16. endif(WIN32)
  17. if (UNIX)
  18. TARGET_LINK_LIBRARIES(YOLOv8 "${ONNXRUNTIME_DIR}/lib/libonnxruntime.so")
  19. endif(UNIX)