From db22498c6c9f4c992d7fc4838e09cf5b150d3de4 Mon Sep 17 00:00:00 2001 From: Francis Zhang <25652281+Francis777@users.noreply.github.com> Date: Mon, 9 Jan 2023 19:23:39 +0000 Subject: [PATCH] fix --- scripts/gen_info.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/scripts/gen_info.py b/scripts/gen_info.py index 4eeb09d..039d086 100644 --- a/scripts/gen_info.py +++ b/scripts/gen_info.py @@ -78,7 +78,7 @@ def generate_info(nusc, scenes, max_cam_sweeps=6, max_lidar_sweeps=10): 'sample_token']: break sweep_cam_info['ego_pose'] = nusc.get( - 'ego_pose', cam_data['ego_pose_token']) + 'ego_pose', sweep_cam_data['ego_pose_token']) sweep_cam_info['timestamp'] = sweep_cam_data[ 'timestamp'] sweep_cam_info['is_key_frame'] = sweep_cam_data[ @@ -88,7 +88,7 @@ def generate_info(nusc, scenes, max_cam_sweeps=6, max_lidar_sweeps=10): sweep_cam_info['filename'] = sweep_cam_data['filename'] sweep_cam_info['calibrated_sensor'] = nusc.get( 'calibrated_sensor', - cam_data['calibrated_sensor_token']) + sweep_cam_data['calibrated_sensor_token']) cam_sweeps[j][cam_names[k]] = sweep_cam_info for k, lidar_data in enumerate(lidar_datas): @@ -115,7 +115,7 @@ def generate_info(nusc, scenes, max_cam_sweeps=6, max_lidar_sweeps=10): 'filename'] sweep_lidar_info['calibrated_sensor'] = nusc.get( 'calibrated_sensor', - cam_data['calibrated_sensor_token']) + sweep_lidar_data['calibrated_sensor_token']) lidar_sweeps[j][lidar_names[k]] = sweep_lidar_info # Remove empty sweeps. for i, sweep in enumerate(cam_sweeps):