CMakeLists.txt 982 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. cmake_minimum_required(VERSION 3.9)
  2. set(CMAKE_CXX_STANDARD 17)
  3. project(picodet_demo)
  4. find_package(OpenMP REQUIRED)
  5. if(OPENMP_FOUND)
  6. message("OPENMP FOUND")
  7. set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
  8. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
  9. set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
  10. endif()
  11. # find_package(OpenCV REQUIRED)
  12. find_package(OpenCV REQUIRED PATHS "/path/to/opencv-3.4.16_gcc8.2_ffmpeg")
  13. # find_package(ncnn REQUIRED)
  14. find_package(ncnn REQUIRED PATHS "/path/to/ncnn/build/install/lib/cmake/ncnn")
  15. if(NOT TARGET ncnn)
  16. message(WARNING "ncnn NOT FOUND! Please set ncnn_DIR environment variable")
  17. else()
  18. message("ncnn FOUND ")
  19. endif()
  20. include_directories(
  21. ${OpenCV_INCLUDE_DIRS}
  22. ${CMAKE_CURRENT_SOURCE_DIR}
  23. ${CMAKE_CURRENT_BINARY_DIR}
  24. )
  25. add_executable(picodet_demo main.cpp picodet.cpp)
  26. target_link_libraries(
  27. picodet_demo
  28. ncnn
  29. ${OpenCV_LIBS}
  30. )