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)