Skip to content

Commit 8981169

Browse files
[Fix] Add num_features in Lyft dataset info file (#1948)
Co-authored-by: Tai-Wang <tab_wang@outlook.com>
1 parent fdb17a5 commit 8981169

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

tools/dataset_converters/lyft_converter.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ def _fill_trainval_infos(lyft,
131131

132132
info = {
133133
'lidar_path': lidar_path,
134+
'num_features': 5,
134135
'token': sample['token'],
135136
'sweeps': [],
136137
'cams': dict(),

tools/dataset_converters/update_infos_to_v2.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -761,6 +761,8 @@ def update_lyft_infos(pkl_path, out_dir):
761761
temp_data_info['ego2global'] = convert_quaternion_to_matrix(
762762
ori_info_dict['ego2global_rotation'],
763763
ori_info_dict['ego2global_translation'])
764+
temp_data_info['lidar_points']['num_pts_feats'] = ori_info_dict.get(
765+
'num_features', 5)
764766
temp_data_info['lidar_points']['lidar_path'] = Path(
765767
ori_info_dict['lidar_path']).name
766768
temp_data_info['lidar_points'][

0 commit comments

Comments
 (0)