From d534fcc0e62e57564e9ed1c1da6534b8a1735713 Mon Sep 17 00:00:00 2001 From: Kazadhum Date: Wed, 22 Jan 2025 17:28:13 +0000 Subject: [PATCH] fix(collect): fixed data blobs in IMU data in collections #1000 A bug was causing the IMU data recorded in the collections (and not in the continuous_sensor_data field) to be recorded as blob. Added message processing to turn it into a dictionary. --- .../src/atom_calibration/collect/data_collector.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/atom_calibration/src/atom_calibration/collect/data_collector.py b/atom_calibration/src/atom_calibration/collect/data_collector.py index 7f2501ef..34ac8622 100644 --- a/atom_calibration/src/atom_calibration/collect/data_collector.py +++ b/atom_calibration/src/atom_calibration/collect/data_collector.py @@ -622,7 +622,9 @@ def saveCollection(self): msg = self.label_msgs[sensor_key].point_cloud elif sensor['modality'] in ['depth']: msg = self.label_msgs[sensor_key].image - + elif sensor['modality'] in ['imu']: + msg = self.label_msgs[sensor_key] # No need to separate the data in the message for IMU + all_sensor_data_dict[sensor['_name'] ] = message_converter.convert_ros_message_to_dictionary(msg)