prepare_dataset.md
July 23, 2024 · View on GitHub
SemanticKITTI
To prepare for SemanticKITTI dataset, please download the KITTI Odometry Dataset (including color, velodyne laser data, and calibration files) and the annotations for Semantic Scene Completion from SemanticKITTI. Put all .zip files under SparseOcc/data/SemanticKITTI and unzip these files. Then you should get the following dataset structure:
SparseOcc
├── data/
│ ├── SemanticKITTI/
│ │ ├── dataset/
│ │ │ ├── sequences
│ │ │ │ ├── 00
│ │ │ │ │ ├── calib.txt
│ │ │ │ │ ├── image_2/
│ │ │ │ │ ├── image_3/
│ │ │ │ │ ├── voxels/
│ │ │ │ ├── 01
│ │ │ │ ├── 02
│ │ │ │ ├── ...
│ │ │ │ ├── 21
Preprocess the annotations for semantic scene completion:
python projects/mmdet3d_plugin/tools/kitti_process/semantic_kitti_preprocess.py --kitti_root data/SemanticKITTI --kitti_preprocess_root data/SemanticKITTI --data_info_path projects/mmdet3d_plugin/tools/kitti_process/semantic-kitti.yaml
OpenOccupancy
Please refer to OpenOccupancy for dataset download and preparation. The final folder structure should be
SparseOcc
├── data/
│ ├── nuscenes/
│ │ ├── maps/
│ │ ├── samples/
│ │ ├── sweeps/
│ │ ├── lidarseg/
│ │ ├── v1.0-test/
│ │ ├── v1.0-trainval/
│ │ ├── nuscenes_occ_infos_train.pkl/
│ │ ├── nuscenes_occ_infos_val.pkl/
│ ├── depth_gt/
│ ├── nuScenes-Occupancy/