@@ -35,9 +35,6 @@ def kitti_data_prep(root_path,
35
35
info_train_path = osp .join (out_dir , f'{ info_prefix } _infos_train.pkl' )
36
36
info_val_path = osp .join (out_dir , f'{ info_prefix } _infos_val.pkl' )
37
37
info_trainval_path = osp .join (out_dir , f'{ info_prefix } _infos_trainval.pkl' )
38
- kitti .export_2d_annotation (root_path , info_train_path )
39
- kitti .export_2d_annotation (root_path , info_val_path )
40
- kitti .export_2d_annotation (root_path , info_trainval_path )
41
38
update_pkl_infos ('kitti' , out_dir = out_dir , pkl_path = info_train_path )
42
39
update_pkl_infos ('kitti' , out_dir = out_dir , pkl_path = info_val_path )
43
40
update_pkl_infos ('kitti' , out_dir = out_dir , pkl_path = info_trainval_path )
@@ -76,17 +73,11 @@ def nuscenes_data_prep(root_path,
76
73
77
74
if version == 'v1.0-test' :
78
75
info_test_path = osp .join (out_dir , f'{ info_prefix } _infos_test.pkl' )
79
- nuscenes_converter .export_2d_annotation (
80
- root_path , info_test_path , version = version )
81
76
update_pkl_infos ('nuscenes' , out_dir = out_dir , pkl_path = info_test_path )
82
77
return
83
78
84
79
info_train_path = osp .join (out_dir , f'{ info_prefix } _infos_train.pkl' )
85
80
info_val_path = osp .join (out_dir , f'{ info_prefix } _infos_val.pkl' )
86
- nuscenes_converter .export_2d_annotation (
87
- root_path , info_train_path , version = version )
88
- nuscenes_converter .export_2d_annotation (
89
- root_path , info_val_path , version = version )
90
81
update_pkl_infos ('nuscenes' , out_dir = out_dir , pkl_path = info_train_path )
91
82
update_pkl_infos ('nuscenes' , out_dir = out_dir , pkl_path = info_val_path )
92
83
create_groundtruth_database (dataset_name , root_path , info_prefix ,
0 commit comments