|
| 1 | +cmake_minimum_required(VERSION 3.16) |
| 2 | + |
| 3 | +project(seg_label_generate) |
| 4 | + |
| 5 | +set(CMAKE_CXX_STANDARD 11) |
| 6 | + |
| 7 | +add_definitions( |
| 8 | + -DCPU_ONLY |
| 9 | +) |
| 10 | + |
| 11 | +set(EXECUTE_COMMAND find src/ -name "*.cpp") |
| 12 | + |
| 13 | +execute_process( |
| 14 | + COMMAND ${EXECUTE_COMMAND} |
| 15 | + WORKING_DIRECTORY ${CMAKE_SOURCE_DIR} |
| 16 | + OUTPUT_VARIABLE CXX_SRCS_RAW |
| 17 | +) |
| 18 | +string(REPLACE "\n" ";" CXX_SRCS ${CXX_SRCS_RAW}) # drop blank line |
| 19 | + |
| 20 | +add_executable(${PROJECT_NAME} |
| 21 | + ${CXX_SRCS} |
| 22 | +) |
| 23 | + |
| 24 | +set(dep_libs "") |
| 25 | + |
| 26 | +#--- OpenCV |
| 27 | +# You may switch different version of OpenCV like this: |
| 28 | +# set(OpenCV_DIR "/usr/local/opencv-4.3.0" CACHE PATH "") |
| 29 | +find_package(OpenCV REQUIRED |
| 30 | + COMPONENTS core highgui imgproc imgcodecs |
| 31 | +) |
| 32 | +if(NOT OpenCV_FOUND) # if not OpenCV 4.x/3.x, then imgcodecs are not found |
| 33 | + find_package(OpenCV REQUIRED COMPONENTS core highgui imgproc) |
| 34 | +endif() |
| 35 | + |
| 36 | +list(APPEND dep_libs |
| 37 | + PUBLIC ${OpenCV_LIBS} |
| 38 | +) |
| 39 | + |
| 40 | +#--- OpenMP |
| 41 | +find_package(OpenMP) |
| 42 | +if(NOT TARGET OpenMP::OpenMP_CXX AND (OpenMP_CXX_FOUND OR OPENMP_FOUND)) |
| 43 | + target_compile_options(${PROJECT_NAME} PRIVATE ${OpenMP_CXX_FLAGS}) |
| 44 | +endif() |
| 45 | + |
| 46 | +if(OpenMP_CXX_FOUND OR OPENMP_FOUND) |
| 47 | + message(STATUS "Building with OpenMP") |
| 48 | + if(OpenMP_CXX_FOUND) |
| 49 | + list(APPEND dep_libs PUBLIC OpenMP::OpenMP_CXX) |
| 50 | + else() |
| 51 | + list(APPEND dep_libs PRIVATE "${OpenMP_CXX_FLAGS}") |
| 52 | + endif() |
| 53 | +endif() |
| 54 | + |
| 55 | +# --- target config with include dirs / libs |
| 56 | +target_link_libraries(${PROJECT_NAME} |
| 57 | + ${dep_libs} |
| 58 | +) |
| 59 | + |
| 60 | +target_include_directories(${PROJECT_NAME} |
| 61 | + PUBLIC include |
| 62 | +) |
0 commit comments