From d0b105deacd774b9e58f511b37d8cd0f935555c7 Mon Sep 17 00:00:00 2001 From: manuelgitgomes Date: Wed, 15 May 2024 10:55:49 +0100 Subject: [PATCH] Fix for depth manual labeler for multi pattern Depth manual labeler was not correctly adapted for the changes done in multi pattern A simple for loop did the trick --- .../atom_calibration/dataset_playback/depth_manual_labeling.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/atom_calibration/src/atom_calibration/dataset_playback/depth_manual_labeling.py b/atom_calibration/src/atom_calibration/dataset_playback/depth_manual_labeling.py index fa623357..94787965 100755 --- a/atom_calibration/src/atom_calibration/dataset_playback/depth_manual_labeling.py +++ b/atom_calibration/src/atom_calibration/dataset_playback/depth_manual_labeling.py @@ -89,7 +89,8 @@ def clickedPointsCallback(point_msg, clicked_points, dataset, sensor_key, select idxs_to_remove.reverse() for idx in idxs_to_remove: - del dataset['collections'][collection_key]['labels'][sensor_key]['idxs_limit_points'][idx] + for pattern_key in dataset['calibration_config']['calibration_patterns'].keys(): + del dataset['collections'][collection_key]['labels'][pattern_key][sensor_key]['idxs_limit_points'][idx] clicked_points[collection_key][sensor_key]['valid_polygon'] = True