@ -19,8 +19,13 @@ endif()
enable_language ( C )
enable_language ( CXX )
set ( ENABLE_CUDA "TRUE" CACHE BOOL "Enable CUDA build" )
set ( CMAKE_CXX_STANDARD 11 )
set ( CMAKE_MODULE_PATH "${CMAKE_CURRENT_LIST_DIR}/cmake/Modules/" ${ CMAKE_MODULE_PATH } )
include_directories ( ${ CMAKE_CURRENT_LIST_DIR } /include )
include_directories ( ${ CMAKE_CURRENT_LIST_DIR } /src )
set ( SKIP_USELIB_TRACK "FALSE" CACHE BOOL "Skip building uselib_track" FORCE )
set ( ENABLE_CUDA "TRUE" CACHE BOOL "Enable CUDA build" )
if ( ${ CMAKE_VERSION } VERSION_LESS "3.8.0" )
set ( ENABLE_CUDA "FALSE" CACHE BOOL "Enable CUDA build" FORCE )
message ( WARNING "To build with CUDA support you need CMake 3.8.0+" )
@ -34,11 +39,6 @@ else()
endif ( )
endif ( )
set ( CMAKE_CXX_STANDARD 11 )
set ( CMAKE_MODULE_PATH "${CMAKE_CURRENT_LIST_DIR}/cmake/Modules/" ${ CMAKE_MODULE_PATH } )
include_directories ( ${ CMAKE_CURRENT_LIST_DIR } /include )
include_directories ( ${ CMAKE_CURRENT_LIST_DIR } /src )
set ( CUDA_COMPUTE_MODEL_75 "75" ) # C o m p u t e c a p a b i l i t y f o r T e s l a T 4 , R T X 2 0 8 0
set ( CUDA_COMPUTE_MODEL_72 "72" ) # C o m p u t e c a p a b i l i t y f o r J e t s o n X a v i e r
set ( CUDA_COMPUTE_MODEL_70 "70" ) # C o m p u t e c a p a b i l i t y f o r T e s l a V 1 0 0
@ -103,6 +103,10 @@ else()
endif ( )
if ( OpenCV_FOUND )
if ( NOT OpenCV_CUDA_VERSION )
set ( SKIP_USELIB_TRACK "TRUE" CACHE BOOL "Skip building uselib_track" FORCE )
message ( WARNING "The build of OpenCV does not have CUDA support, so some tools have been disabled! Please rebuild OpenCV with CUDA support to build them!" )
endif ( )
add_definitions ( -DOPENCV )
include_directories ( ${ OpenCV_INCLUDE_DIRS } )
endif ( )
@ -227,7 +231,7 @@ endif()
target_compile_definitions ( darklib PRIVATE LIB_EXPORTS=1 )
set_target_properties ( darklib PROPERTIES LINKER_LANGUAGE CXX )
if ( OpenCV_VERSION VERSION_GREATER "3.0" )
if ( OpenCV_VERSION VERSION_GREATER "3.0" AND NOT SKIP_USELIB_TRACK )
add_executable ( uselib_track ${ CMAKE_CURRENT_LIST_DIR } /src/yolo_console_dll.cpp )
target_compile_definitions ( uselib_track PRIVATE TRACK_OPTFLOW=1 )
set_target_properties ( uselib_track PROPERTIES LINKER_LANGUAGE CXX )
@ -276,7 +280,7 @@ if(ENABLE_CUDA)
endif ( )
target_link_libraries ( uselib darklib )
if ( OpenCV_VERSION VERSION_GREATER "3.0" )
if ( OpenCV_VERSION VERSION_GREATER "3.0" AND NOT SKIP_USELIB_TRACK )
target_link_libraries ( uselib_track darklib )
install ( TARGETS uselib_track DESTINATION ${ CMAKE_SOURCE_DIR } / )
endif ( )