faster-rcnn-cpp/CMakeLists.txt at main · ReadWriteV/faster-rcnn-cpp · 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
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)

project(faster_rcnn CXX)

find_package(Torch REQUIRED)
find_package(TorchVision REQUIRED)
find_package(Boost REQUIRED COMPONENTS program_options json headers)
find_package(OpenCV REQUIRED)

add_subdirectory(utils)
add_subdirectory(dataset)
add_subdirectory(model_r50_fpn)
add_subdirectory(model_vgg)
add_subdirectory(eval)

add_executable(train train.cpp)
set_target_properties(train PROPERTIES CXX_STANDARD 17)
target_link_libraries(train PRIVATE model_vgg dataset Boost::program_options Boost::json)

add_executable(test test.cpp)
set_target_properties(test PROPERTIES CXX_STANDARD 17)
target_link_libraries(test PRIVATE model_vgg dataset Boost::program_options Boost::json)

add_executable(train_r50fpn train_r50fpn.cpp)
set_target_properties(train_r50fpn PROPERTIES CXX_STANDARD 17)
target_link_libraries(train_r50fpn PRIVATE model_r50_fpn Boost::program_options Boost::headers)

add_executable(test_r50fpn test_r50fpn.cpp)
set_target_properties(test_r50fpn PROPERTIES CXX_STANDARD 17)
target_link_libraries(test_r50fpn PRIVATE model_r50_fpn Boost::program_options Boost::headers)