diff --git a/tools/create_data.py b/tools/create_data.py index 08d210e58..f2012d931 100644 --- a/tools/create_data.py +++ b/tools/create_data.py @@ -35,9 +35,6 @@ def kitti_data_prep(root_path, info_train_path = osp.join(out_dir, f'{info_prefix}_infos_train.pkl') info_val_path = osp.join(out_dir, f'{info_prefix}_infos_val.pkl') info_trainval_path = osp.join(out_dir, f'{info_prefix}_infos_trainval.pkl') - kitti.export_2d_annotation(root_path, info_train_path) - kitti.export_2d_annotation(root_path, info_val_path) - kitti.export_2d_annotation(root_path, info_trainval_path) update_pkl_infos('kitti', out_dir=out_dir, pkl_path=info_train_path) update_pkl_infos('kitti', out_dir=out_dir, pkl_path=info_val_path) update_pkl_infos('kitti', out_dir=out_dir, pkl_path=info_trainval_path) @@ -76,17 +73,11 @@ def nuscenes_data_prep(root_path, if version == 'v1.0-test': info_test_path = osp.join(out_dir, f'{info_prefix}_infos_test.pkl') - nuscenes_converter.export_2d_annotation( - root_path, info_test_path, version=version) update_pkl_infos('nuscenes', out_dir=out_dir, pkl_path=info_test_path) return info_train_path = osp.join(out_dir, f'{info_prefix}_infos_train.pkl') info_val_path = osp.join(out_dir, f'{info_prefix}_infos_val.pkl') - nuscenes_converter.export_2d_annotation( - root_path, info_train_path, version=version) - nuscenes_converter.export_2d_annotation( - root_path, info_val_path, version=version) update_pkl_infos('nuscenes', out_dir=out_dir, pkl_path=info_train_path) update_pkl_infos('nuscenes', out_dir=out_dir, pkl_path=info_val_path) create_groundtruth_database(dataset_name, root_path, info_prefix,