diff --git a/scripts/validation_dataset/create_validation_dataset.py b/scripts/validation_dataset/create_validation_dataset.py index ccd204d..ad8b6d9 100644 --- a/scripts/validation_dataset/create_validation_dataset.py +++ b/scripts/validation_dataset/create_validation_dataset.py @@ -97,7 +97,6 @@ def generate(self): topics = ['/image_raw', '/radar/cfar_detections', '/tf_static'] points_radar_window = [] snr_radar_window = [] - noise_radar_window = [] bridge = CvBridge() points_radar_all = None snr_radar_all = None @@ -106,7 +105,7 @@ def generate(self): for i, (topic, msg, t) in enumerate(bag.read_messages(topics=topics)): if topic == '/radar/cfar_detections': # Transform PC to camera frame - points_radar, snr_radar, noise_radar = self.pointcloud2_to_xyz_array(msg) + points_radar, snr_radar, _ = self.pointcloud2_to_xyz_array(msg) points_radar = points_radar.T noise_radar = noise_radar.T @@ -385,6 +384,7 @@ def process_camera_data(self, data_dict, points): def get_depth_dict(self): + # Prepare a multiprocessing pool pool = mp.Pool(mp.cpu_count()) # Arguments for each process