1 cmake_minimum_required(VERSION 2.8)
3 set(KCF_LIB_SRC kcf.cpp kcf.h fft.cpp)
5 SET(FFT "OpenCV" CACHE STRING "Select FFT implementation")
6 SET_PROPERTY(CACHE FFT PROPERTY STRINGS OpenCV OpenCV_cuFFT fftw cuFFTW cuFFT)
7 MESSAGE(STATUS "FFT implementation: ${FFT}")
9 option(OPENMP "Use OpenMP library. Works with FFTW and OpenCV implementation." OFF)
10 option(ASYNC "Works only if OPENCV_CUFFT is not ON. Will enable C++ async directive." OFF)
11 option(PROFILING "Enable profiling using perf_event_open together with libpfm4. " OFF)
12 option(BIG_BATCH "Enable transforming all features from all scales together." OFF)
15 add_definitions(-DPROFILING )
16 MESSAGE(STATUS "Profiling mode")
20 add_definitions(-DBIG_BATCH )
21 MESSAGE(STATUS "Big_batch mode")
26 IF(FFT STREQUAL "OpenCV")
27 list(APPEND KCF_LIB_SRC fft_opencv.cpp complexmat.hpp)
28 ELSEIF(FFT STREQUAL "OpenCV_cuFFT")
29 list(APPEND KCF_LIB_SRC fft_opencv_cuda.cpp complexmat.hpp)
30 add_definitions(-DOPENCV_CUFFT)
32 ELSEIF(FFT STREQUAL "fftw")
33 list(APPEND KCF_LIB_SRC fft_fftw.cpp complexmat.hpp)
34 add_definitions(-DFFTW)
35 ELSEIF(FFT STREQUAL "cuFFTW")
36 list(APPEND KCF_LIB_SRC fft_fftw.cpp complexmat.hpp)
37 add_definitions(-DFFTW -DCUFFTW)
39 ELSEIF(FFT STREQUAL "cuFFT")
40 list(APPEND KCF_LIB_SRC fft_cufft.cpp complexmat.cuh)
41 add_definitions(-DCUFFT)
44 MESSAGE(FATAL_ERROR "Invalid FFT implementation selected")
47 IF(ASYNC AND NOT OPENMP)
48 add_definitions(-DASYNC)
49 MESSAGE(STATUS "ASYNC")
52 FIND_PACKAGE( OpenCV REQUIRED )
55 find_package(CUDA REQUIRED)
57 if (NOT ${OpenCV_USE_CUFFT})
58 message(SEND_ERROR "OpenCV was not built with cuFFT")
61 include_directories ( ${CUDA_INCLUDE_DIRS} )
62 set(CUDA_SEPARABLE_COMPILATION ON)
63 set(CUDA_PROPAGATE_HOST_FLAGS OFF)
64 set(CUDA_HOST_COMPILER /usr/bin/g++)
65 list( APPEND CUDA_NVCC_FLAGS -O3 --gpu-architecture sm_62 -std=c++11 -default-stream per-thread)
66 find_cuda_helper_libs(cufftw)
67 IF(FFT STREQUAL "cuFFT")
68 cuda_add_library(complexmat complexmat.cu)
73 add_subdirectory(piotr_fhog)
76 add_library(kcf STATIC ${KCF_LIB_SRC})
77 target_link_libraries(kcf fhog cndata ${OpenCV_LIBS})
78 set_target_properties(kcf PROPERTIES VERSION 1.0.0 SOVERSION 1)
80 IF(FFT STREQUAL "fftw")
81 target_link_libraries(kcf fftw3f)
83 target_link_libraries(kcf fftw3_omp)
85 target_link_libraries(kcf fftw3_threads)
89 IF(FFT STREQUAL "cuFFTW")
90 target_link_libraries(kcf ${CUDA_cufft_LIBRARY} ${CUDA_cufftw_LIBRARY})
93 IF(FFT STREQUAL "cuFFT")
94 target_link_libraries(kcf ${CUDA_cufft_LIBRARY} ${CUDA_LIBRARIES} complexmat)
98 target_link_libraries(kcf pfm)