prepare_dataset.md

September 16, 2024 · View on GitHub

Prepare Dataset

NuScenes

1. Download nuScenes V1.0 dataset

2. Link dataset to project

ln -s path/to/nuscenes ./data/nuscenes

3. Convert nuscenes dataset

python tools/data_converter/nuscenes_converter.py --info_prefix ./data/nuscenes/simpb_nuscenes

Kmean Anchors

python tools/anchor_generator.py --ann_file ./data/nuscenes/simpb_nuscenes_infos_train.pkl

Folder structure

SimPB
├── projects/
├── tools/
├── ckpts/
├── data/
│   ├── nuscenes/
│   │   ├── maps/
│   │   ├── samples/
│   │   ├── sweeps/
│   │   ├── v1.0-test/
|   |   ├── v1.0-trainval/
|   |   ├── nuscenes_kmeans900.npy
|   |   ├── simpb_nuscenes_infos_test.pkl
|   |   ├── simpb_nuscenes_infos_train.pkl
|   |   ├── simpb_nuscenes_infos_val.pkl