Prepare data related to Kitti dataset. Related data consists of '.pkl' files recording basic infos, 2D annotations and groundtruth database. Args: root_path (str): Path of dataset root. info_prefix (str): The prefix of info filenames. version (str): Dataset vers
(root_path, info_prefix, version, out_dir)
| 9 | |
| 10 | |
| 11 | def kitti_data_prep(root_path, info_prefix, version, out_dir): |
| 12 | """Prepare data related to Kitti dataset. |
| 13 | |
| 14 | Related data consists of '.pkl' files recording basic infos, |
| 15 | 2D annotations and groundtruth database. |
| 16 | |
| 17 | Args: |
| 18 | root_path (str): Path of dataset root. |
| 19 | info_prefix (str): The prefix of info filenames. |
| 20 | version (str): Dataset version. |
| 21 | out_dir (str): Output directory of the groundtruth database info. |
| 22 | """ |
| 23 | kitti.create_kitti_info_file(root_path, info_prefix) |
| 24 | kitti.create_reduced_point_cloud(root_path, info_prefix) |
| 25 | create_groundtruth_database( |
| 26 | 'KittiDataset', |
| 27 | root_path, |
| 28 | info_prefix, |
| 29 | f'{out_dir}/{info_prefix}_infos_train.pkl', |
| 30 | relative_path=False, |
| 31 | mask_anno_path='instances_train.json', |
| 32 | with_mask=(version == 'mask')) |
| 33 | |
| 34 | |
| 35 | def nuscenes_data_prep(root_path, |
no test coverage detected