| cmake_minimum_required(VERSION 3.18 FATAL_ERROR) | |
| project(yolov8_libtorch_example) | |
| set(CMAKE_CXX_STANDARD 17) | |
| set(CMAKE_CXX_STANDARD_REQUIRED ON) | |
| set(CMAKE_CXX_EXTENSIONS OFF) | |
| # -------------- OpenCV -------------- | |
| set(OpenCV_DIR "/path/to/opencv/lib/cmake/opencv4") | |
| find_package(OpenCV REQUIRED) | |
| message(STATUS "OpenCV library status:") | |
| message(STATUS " config: ${OpenCV_DIR}") | |
| message(STATUS " version: ${OpenCV_VERSION}") | |
| message(STATUS " libraries: ${OpenCV_LIBS}") | |
| message(STATUS " include path: ${OpenCV_INCLUDE_DIRS}") | |
| include_directories(${OpenCV_INCLUDE_DIRS}) | |
| # -------------- libtorch -------------- | |
| list(APPEND CMAKE_PREFIX_PATH "/path/to/libtorch") | |
| set(Torch_DIR "/path/to/libtorch/share/cmake/Torch") | |
| find_package(Torch REQUIRED) | |
| set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}") | |
| message("${TORCH_LIBRARIES}") | |
| message("${TORCH_INCLUDE_DIRS}") | |
| # The following code block is suggested to be used on Windows. | |
| # According to https://github.com/pytorch/pytorch/issues/25457, | |
| # the DLLs need to be copied to avoid memory errors. | |
| # if (MSVC) | |
| # file(GLOB TORCH_DLLS "${TORCH_INSTALL_PREFIX}/lib/*.dll") | |
| # add_custom_command(TARGET yolov8_libtorch_example | |
| # POST_BUILD | |
| # COMMAND ${CMAKE_COMMAND} -E copy_if_different | |
| # ${TORCH_DLLS} | |
| # $<TARGET_FILE_DIR:yolov8_libtorch_example>) | |
| # endif (MSVC) | |
| include_directories(${TORCH_INCLUDE_DIRS}) | |
| add_executable(yolov8_libtorch_inference "${CMAKE_CURRENT_SOURCE_DIR}/main.cc") | |
| target_link_libraries(yolov8_libtorch_inference ${TORCH_LIBRARIES} ${OpenCV_LIBS}) | |
| set_property(TARGET yolov8_libtorch_inference PROPERTY CXX_STANDARD 17) | |