data_preparation.md

August 7, 2023 · View on GitHub

You can download nuScenes 3D detection data HERE and unzip all zip files. You can only download v1.0-trainval version conveniently. Like the general way to prepare dataset, it is recommended to symlink the dataset root to $SOGDet/data.

We typically need to organize the useful data information with a .pkl. To prepare these files for nuScenes, run the following command:

python tools/create_data.py  
python tools/data_converter/prepare_nuscenes_for_bevdet4d.py  

Then download nuscenes panoptic dataset HERE. Symlink the dataset root to $SOGDet/data/nuscenes and replace .json file following HERE. Run the following command to generate point cloud with panoptic labels.

python tools/data_converter/prepare_panoptic.py

Here, the total data structure is shown like this:

mmdetection3d
├── mmdet3d
├── tools
├── configs
├── data
│   ├── nuscenes
│   │   ├── maps
│   │   ├── panoptic
│   │   ├── pc_panoptic
│   │   ├── samples
│   │   ├── sweeps
│   │   ├── v1.0-test
|   |   ├── v1.0-trainval
│   │   ├── nuscenes_infos_train.pkl
│   │   ├── nuscenes_infos_val.pkl
│   │   ├── nuscenes_infos_train_4d_interval3_max60.pkl
│   │   ├── nuscenes_infos_val_4d_interval3_max60.pkl