Skip to content

Latest commit

 

History

History
38 lines (28 loc) · 1.06 KB

prepare_dataset.md

File metadata and controls

38 lines (28 loc) · 1.06 KB

Preparing Dataset

  1. Organize your folder structure as below:

    Downloading annotations.json

UniScene
├── projects/
├── tools/
├── ckpts/
│   ├── r101_dcn_fcos3d_pretrain.pth
├── data/
│   ├── can_bus/
│   ├── nuscenes/
│   │   ├── maps/
│   │   ├── samples/     
│   │   ├── sweeps/     
│   │   ├── v1.0-trainval/
│   │   ├── gts/
│   │   └── annotations.json
  1. Generate the binary geometry occupancy labels to gts

    or download from gts

python tools/creat_binary_occ_labels.py --dataroot ./data/nuscenes/ --save_path ./data/nuscenes/ --num_sweeps 2
  1. Generate the info files for training and validation:
python tools/create_data.py occ --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag occ --version v1.0-trainval --canbus ./data --occ-path ./data/nuscenes