CMakeLists.txt的设置如下即可,注意将Torch_DIR的路径更换为自己的libtorch存放的路径.
cmake_minimum_required(VERSION 3.5)
project(libtorch-yolov5)
set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall")
find_package(OpenCV REQUIRED)
set(Torch_DIR /home/think/libtorch)
find_package(Torch PATHS ${Torch_DIR} NO_DEFAULT REQUIRED)
add_executable(${CMAKE_PROJECT_NAME} ${SOURCE_FILES})
target_link_libraries (
${CMAKE_PROJECT_NAME}
${OpenCV_LIBS}
${TORCH_LIBRARIES}
)