Data Preparation
October 10, 2024 · View on GitHub
Dataset
1. Download nuScenes
Download the nuScenes dataset to ./data/nuscenes.
2. Creating infos file
We modify data preparation in MMDetection3D, which addtionally creates 2D annotations and temporal information for training/evaluation.
python tools/create_data_nusc.py --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes2d --version v1.0
Using the above code will generate nuscenes2d_temporal_infos_{train,val}.pkl.
We also privided the processed train, val and test pkl.
- After preparation, you will be able to see the following directory structure:
Folder structure
QAF2D
├── projects/
├── mmdetection3d/
├── tools/
├── data/
│ ├── nuscenes/
│ │ ├── maps/
│ │ ├── samples/
│ │ ├── sweeps/
│ │ ├── v1.0-test/
| | ├── v1.0-trainval/
| | ├── nuscenes2d_temporal_infos_train.pkl
| | ├── nuscenes2d_temporal_infos_val.pkl
In addition, you also need to rename nuscenes2d_temporal_infos_train.pkl to nuscenes2d_temporal_infos_train_stream.pkl and rename nuscenes2d_temporal_infos_val.pkl to nuscenes2d_temporal_infos_val_stream.pkl