-
Notifications
You must be signed in to change notification settings - Fork 0
/
CMakeLists.txt
90 lines (76 loc) · 3.36 KB
/
CMakeLists.txt
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
cmake_minimum_required(VERSION 2.8.12)
project(ros_chainercv)
# Make sure to find-package `catkin_virtualenv`
find_package(catkin REQUIRED catkin_virtualenv)
catkin_python_setup()
# Must be called before catkin_generate_virtualenv
catkin_package()
# Generate the virtualenv, optionally with python 3 as the default interpreter:
catkin_generate_virtualenv(
INPUT_REQUIREMENTS requirements.in
PYTHON_INTERPRETER python2.7
USE_SYSTEM_PACKAGES TRUE
ISOLATE_REQUIREMENTS FALSE
CHECK_VENV FALSE
)
# catkin_generate_virtualenv(PYTHON3)
file(GLOB NODE_SCRIPTS_FILES node_scripts/*)
catkin_install_python(
PROGRAMS ${NODE_SCRIPTS_FILES}
DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
)
catkin_install_python(
PROGRAMS scripts/install_pretrained_models.py
DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
)
install(
DIRECTORY launch sample test yaml
DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}
USE_SOURCE_PERMISSIONS
)
install(
FILES requirements.in requirements.txt
DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}
)
add_custom_target(${PROJECT_NAME}_install_pretrained_models ALL
COMMAND ${${PROJECT_NAME}_VENV_DEVEL_DIR}/bin/python ${PROJECT_SOURCE_DIR}/scripts/install_pretrained_models.py
DEPENDS ${PROJECT_NAME}_generate_virtualenv
)
if(CATKIN_ENABLE_TESTING)
find_package(rostest REQUIRED)
find_package(roslint REQUIRED)
find_package(roslaunch REQUIRED)
find_package(jsk_tools REQUIRED)
roslint_python()
macro(virtualenv_add_rostest rostest_file)
add_rostest(${rostest_file}
DEPENDENCIES
${PROJECT_NAME}_generate_virtualenv
${PROJECT_NAME}_install_pretrained_models
)
roslaunch_add_file_check(${rostest_file})
endmacro()
# comment out slow tests
# virtualenv_add_rostest(test/test_deeplab_semantic_segmentation_node.test)
# virtualenv_add_rostest(test/test_deeplab_v3plus_xception65_semantic_segmentation.test)
# virtualenv_add_rostest(test/test_faster_rcnn_fpn_resnet101_object_detection.test)
# virtualenv_add_rostest(test/test_faster_rcnn_fpn_resnet50_object_detection.test)
# virtualenv_add_rostest(test/test_faster_rcnn_fpn_object_detection_node.test)
# virtualenv_add_rostest(test/test_faster_rcnn_vgg16_object_detection.test)
# virtualenv_add_rostest(test/test_faster_rcnn_object_detection_node.test)
# virtualenv_add_rostest(test/test_fcis_resnet101_instance_segmentation.test)
# virtualenv_add_rostest(test/test_fcis_instance_segmentation_node.test)
# virtualenv_add_rostest(test/test_mask_rcnn_fpn_resnet101_instance_segmentation.test)
# virtualenv_add_rostest(test/test_mask_rcnn_fpn_resnet50_instance_segmentation.test)
# virtualenv_add_rostest(test/test_mask_rcnn_fpn_instance_segmentation_node.test)
# virtualenv_add_rostest(test/test_pspnet_resnet101_semantic_segmentation.test)
# virtualenv_add_rostest(test/test_pspnet_resnet50_semantic_segmentation.test)
# virtualenv_add_rostest(test/test_pspnet_semantic_segmentation_node.test)
virtualenv_add_rostest(test/test_ssd300_object_detection.test)
# virtualenv_add_rostest(test/test_ssd512_object_detection.test)
virtualenv_add_rostest(test/test_ssd_object_detection_node.test)
virtualenv_add_rostest(test/test_yolo_v2_object_detection.test)
virtualenv_add_rostest(test/test_yolo_v2_tiny_object_detection.test)
virtualenv_add_rostest(test/test_yolo_v3_object_detection.test)
virtualenv_add_rostest(test/test_yolo_object_detection_node.test)
endif()