You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
1.1 KiB
32 lines
1.1 KiB
#!/usr/bin/env python3
|
|
|
|
import cv2
|
|
import sys
|
|
import rospy
|
|
import body_landmark_detector_python3 as bl
|
|
import detection_node_python3 as dn
|
|
import deep_learning_model_options as do
|
|
|
|
if __name__ == '__main__':
|
|
print('cv2.__version__ =', cv2.__version__)
|
|
print('Python version (must be > 3.0):', sys.version)
|
|
assert(int(sys.version[0]) >= 3)
|
|
|
|
models_directory = do.get_directory()
|
|
print('Using the following directory for deep learning models:', models_directory)
|
|
use_neural_compute_stick = do.use_neural_compute_stick()
|
|
|
|
detector = bl.BodyLandmarkDetector(models_directory,
|
|
use_neural_compute_stick=use_neural_compute_stick)
|
|
|
|
default_marker_name = 'body_landmarks'
|
|
node_name = 'DetectBodyLandmarksNode'
|
|
topic_base_name = 'body_landmarks'
|
|
fit_plane = True
|
|
node = dn.DetectionNode(detector, default_marker_name, node_name, topic_base_name, fit_plane)
|
|
node.main()
|
|
try:
|
|
rospy.spin()
|
|
except KeyboardInterrupt:
|
|
print('interrupt received, so shutting down')
|
|
|