Skip to content

Commit 7d7042e

Browse files
authored
Add Orbbec support to process.py (#6)
1 parent f2985c6 commit 7d7042e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

python/cli/process/process.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def define_args(parser):
1414
parser.add_argument("--distance_quantile", help="Max point distance filter quantile (0 = disabled)", type=float, default=0.99)
1515
parser.add_argument("--key_frame_distance", help="Minimum distance between keyframes (meters)", type=float, default=0.05)
1616
parser.add_argument('--no_icp', action='store_true')
17-
parser.add_argument('--device_preset', choices=['none', 'oak-d', 'k4a', 'realsense', 'android-tof', 'ios-tof'], help="Automatically detected in most cases")
17+
parser.add_argument('--device_preset', choices=['none', 'oak-d', 'k4a', 'realsense', 'android-tof', 'ios-tof', 'orbbec-astra2', 'orbbec-femto'], help="Automatically detected in most cases")
1818
parser.add_argument('--fast', action='store_true', help='Fast but lower quality settings')
1919
parser.add_argument('--mono', action='store_true', help='Monocular mode: disable ToF and stereo data')
2020
parser.add_argument('--image_format', type=str, default='jpg', help="Color image format (use 'png' for top quality)")
@@ -467,6 +467,8 @@ def detect_device_preset(input_dir):
467467
if "oak-d" in line: device = "oak-d"
468468
if "k4a" in line: device = "k4a"
469469
if "realsense" in line: device = "realsense"
470+
if "orbbec-astra2" in line: device = "orbbec-astra2"
471+
if "orbbec-femto" in line: device = "orbbec-femto"
470472
if device: break
471473
return (device, cameras)
472474

@@ -525,6 +527,10 @@ def detect_device_preset(input_dir):
525527
elif device_preset == 'oak-d':
526528
config['stereoPointCloudMinDepth'] = 0.5
527529
config['stereoPointCloudStride'] = 30
530+
elif "orbbec" in device_preset:
531+
if prefer_icp:
532+
parameter_sets.extend(['icp'])
533+
if not args.fast: parameter_sets.append('offline-icp')
528534

529535
if args.preview3d:
530536
from spectacularAI.cli.visualization.visualizer import Visualizer

0 commit comments

Comments
 (0)