Skip to content

Latest commit

 

History

History
99 lines (86 loc) · 3.68 KB

prepare_dataset.md

File metadata and controls

99 lines (86 loc) · 3.68 KB

Prepare nuScenes 3D Detection and 3D Occupancy Data

1. Download nuScenes V1.0 full dataset and can bus data HERE. Organize the folder structure:

OccNet
├── data/
│   ├── can_bus/
│   ├── nuscenes/
│   │   ├── maps/
│   │   ├── samples/
│   │   ├── sweeps/
│   │   ├── v1.0-test
│   │   ├── v1.0-trainval

2. Prepared nuScenes 3D detection data

python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes --version v1.0 --canbus ./data

Using the above code will generate the following files data/nuscenes/nuscenes_infos_temporal_{train,val}.pkl

3. Prepare 3D Occupancy dataset

We provide the 3D occupancy and flow annotations for the train and validation dataset. The annotations is defined in the LiDAR cooridinate system including 16 classes.

Dataset Information

Type Info
train | val 28130 | 6019
annotations occupancy | flow
cameras 6
range [-50m, -50m, -5m, 50m, 50m, 3m]
volume size [200, 200, 16]
#classes 0 - 15

1.Dowload train and validataion dataset and put it in the data folder

Version voxel size Google Drive Google Drive Baidu Cloud Baidu Yun Size
occ_gt_release_v1_0 0.5m train_val train_val ~15G

2.unzip the file

tar -zxvf occ_gt_release_v1_0.tar.gz

You will obtain the folder structure

OccNet
├── data/
│   ├── occ_gt_release_v1_0/
│   │   ├── train/
│   │   ├── val/
│   │   ├── occ_gt_train.json
│   │   ├── occ_gt_val.json

3.Merge 3D detection and 3D occupancy dataset

python tools/create_data_with_occ.py

Using the above code will generate the following files data/occ_gt_release_v1_0/nuscenes_infos_temporal_{train,val}_occ_gt.pkl

We also provide the downlink of theses pkls.

Version Google Drive Google Drive Baidu Cloud Baidu Yun
occ_gt_release_v1_0 train | val train | val

4.The data structure of the project is organized as:

OccNet
├── data/
│   ├── can_bus/
│   ├── nuscenes/
│   │   ├── maps/
│   │   ├── samples/
│   │   ├── sweeps/
│   │   ├── v1.0-test
│   │   ├── v1.0-trainval
│   │   ├── nuscenes_infos_temporal_train.pkl
│   │   ├── nuscenes_infos_temporal_val.pkl   
│   ├── occ_gt_release_v1_0/
│   │   ├── train/
│   │   ├── val/
│   │   ├── occ_gt_train.json
│   │   ├── occ_gt_val.json
│   │   ├── nuscenes_infos_temporal_train_occ_gt.pkl
│   │   ├── nuscenes_infos_temporal_val_occ_gt.pkl