RapidSIFT/CMakeLists.txt at main · lowsfer/RapidSIFT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
cmake_minimum_required(VERSION 3.8)
project(RapidSIFT LANGUAGES CXX CUDA)

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CUDA_STANDARD 17)
set(CMAKE_CUDA_ARCHITECTURES 75-real 86-real)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)


include_directories(${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES} ${CMAKE_SOURCE_DIR}/../cuda-utils)
link_directories(${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}/../lib64
        /opt/TensorRT/lib
        ${CMAKE_CURRENT_BINARY_DIR}/../cuda-utils)


option(RAPIDSIFT_ENABLE_SOSNET "The licence of upstream SOSNet weights (https://github.com/scape-research/SOSNet) is unspecified. Use it at your own risk. You have been warned!" OFF)
if(RAPIDSIFT_ENABLE_SOSNET)
        include_directories(${CMAKE_SOURCE_DIR}/../sosnet-ptq)
        link_directories(${CMAKE_SOURCE_DIR}/../sosnet-ptq/build/vscode/)
        add_definitions(-DRAPIDSIFT_ENABLE_SOSNET=1)
else()
        add_definitions(-DRAPIDSIFT_ENABLE_SOSNET=0)
endif()

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=native -Wreturn-type -Wall -Wno-unknown-pragmas -Wextra -Werror")
#set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=native")


find_package(Eigen3 REQUIRED)

set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -lineinfo -keep --expt-relaxed-constexpr") #-res-usage
set(CUDA_PTXAS_FLAGS_RELEASE "${CUDA_PTXAS_FLAGS_RELEASE} -warn-lmem-usage -warn-spills -Werror") # -v
set(CMAKE_CUDA_FLAGS_RELEASE "${CMAKE_CUDA_FLAGS_RELEASE} --use_fast_math -DNDEBUG -Xptxas='${CUDA_PTXAS_FLAGS_RELEASE}'")

#find_library( NVTX_LIBRARY nvToolsExt PATHS ENV LD_LIBRARY_PATH /opt/cuda/lib64)

set(SOURCE_FILES
        RapidSift.h RapidSift.cpp
        conv1d.h types.h
        utils_sift.h
        kernel_DoG.cu
        kernels.h
        kernels.cpp
        TexObj.h
        Pitched2DTexture.h
        utils_host.h
        utils.cpp
        Octave.h Octave.cpp
        SIFT_worker.h SIFT_worker.cpp
        cuda_async_copy_engine.h
        kernel_find_keypoints.cu
        kernel_assign_orientation.cu
        kernel_describe.cu
        kernel_makePatch.cu
        solve_GaussElim.h
        fp16.h
        public_types.h
        sift_master.h
        sift_master.cpp
        ThreadPool.h
        AdaptiveNMS.cu
        KArray.h AdaptiveNMS.h AdaptiveNMS.cpp)

#cuda_add_executable(texture_test texture_test.cu)

add_executable(tests_rapidsift
        tests/test_main.cpp
        tests/DoG_test.cpp
        tests/solve_GaussElim_test.cpp
        tests/find_keypoints_test.cpp
        tests/AdaptiveNMSTest.cpp)
target_compile_definitions(tests_rapidsift PRIVATE BOOST_TEST_DYN_LINK)
target_link_libraries(tests_rapidsift
#        ${CUDA_LIBRARIES}
        boost_unit_test_framework
        RapidSIFT
        cuda-utils
        opencv_core opencv_imgproc opencv_highgui opencv_imgcodecs opencv_highgui
        turbojpeg jpeg heif exiv2
        )

add_library(RapidSIFT ${SOURCE_FILES})
target_link_libraries(RapidSIFT PRIVATE
        cuda-utils)
if(RAPIDSIFT_ENABLE_SOSNET)
        target_link_libraries(RapidSIFT PRIVATE SOSNet_runtime sosnet_rc
        nvinfer)
endif()

find_package(OpenCV)
add_executable(demo_sift main.cpp RapidSift.h filter/robust_gms_matcher.cpp filter/robust_gms_matcher.h
        filter/geometry.cpp filter/PropagationMatchFilter.cpp)
target_link_libraries(demo_sift
        RapidSIFT
        opencv_core opencv_imgproc opencv_highgui opencv_imgcodecs opencv_features2d
        opencv_calib3d
        nvToolsExt
        )