From 53ac268209acfc2de134fdd43d6665b050d9a445 Mon Sep 17 00:00:00 2001 From: wwqgtxx Date: Mon, 21 Dec 2020 11:27:57 +0800 Subject: [PATCH] fix yolov3 build error (#335) --- yolov3/CMakeLists.txt | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/yolov3/CMakeLists.txt b/yolov3/CMakeLists.txt index c7b0db0..03407ee 100644 --- a/yolov3/CMakeLists.txt +++ b/yolov3/CMakeLists.txt @@ -19,14 +19,14 @@ link_directories(/usr/local/cuda/lib64) include_directories(/usr/include/x86_64-linux-gnu/) link_directories(/usr/lib/x86_64-linux-gnu/) +find_package(OpenCV) +include_directories(OpenCV_INCLUDE_DIRS) + set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Ofast -Wfatal-errors -D_MWAITXINTRIN_H_INCLUDED") #cuda_add_library(leaky ${PROJECT_SOURCE_DIR}/leaky.cu) cuda_add_library(yololayer SHARED ${PROJECT_SOURCE_DIR}/yololayer.cu) -target_link_libraries(yololayer nvinfer cudart) - -find_package(OpenCV) -include_directories(OpenCV_INCLUDE_DIRS) +target_link_libraries(yololayer nvinfer cudart ${OpenCV_LIBS}) add_executable(yolov3 ${PROJECT_SOURCE_DIR}/calibrator.cpp ${PROJECT_SOURCE_DIR}/yolov3.cpp) target_link_libraries(yolov3 nvinfer)