Skip to content

GitLab

Commit e9149550 authored by Xiangxu-0103's avatar Xiangxu-0103 Committed by ZwwWayne
Browse files

[Fix] Add num_features in Lyft dataset info file (#1948) 


Co-authored-by: default avatarTai-Wang <tab_wang@outlook.com>
parent ede10946
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
tools/dataset_converters/lyft_converter.py
View file @ e9149550
...@@ -131,6 +131,7 @@ def _fill_trainval_infos(lyft, ...@@ -131,6 +131,7 @@ def _fill_trainval_infos(lyft,
info = { info = {
'lidar_path': lidar_path, 'lidar_path': lidar_path,
'num_features': 5,
'token': sample['token'], 'token': sample['token'],
'sweeps': [], 'sweeps': [],
'cams': dict(), 'cams': dict(),
......
tools/dataset_converters/update_infos_to_v2.py
View file @ e9149550
...@@ -761,6 +761,8 @@ def update_lyft_infos(pkl_path, out_dir): ...@@ -761,6 +761,8 @@ def update_lyft_infos(pkl_path, out_dir):
temp_data_info['ego2global'] = convert_quaternion_to_matrix( temp_data_info['ego2global'] = convert_quaternion_to_matrix(
ori_info_dict['ego2global_rotation'], ori_info_dict['ego2global_rotation'],
ori_info_dict['ego2global_translation']) ori_info_dict['ego2global_translation'])
temp_data_info['lidar_points']['num_pts_feats'] = ori_info_dict.get(
'num_features', 5)
temp_data_info['lidar_points']['lidar_path'] = Path( temp_data_info['lidar_points']['lidar_path'] = Path(
ori_info_dict['lidar_path']).name ori_info_dict['lidar_path']).name
temp_data_info['lidar_points'][ temp_data_info['lidar_points'][
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment