Dataset Processing Tool

The toolchain provides the tools/create_data.py tool for preprocessing of the KITTI-3D and nuScenes datasets.

Usage

Run tools/create_data.py, and the target dataset can be preprocessed accordingly.

Parameters Introduction

ParameterDescription
--datasetDataset Name, the available values are KITTI-3D and nuScenes.
--root-dirThe root directory of the dataset.
--extra-tagAn optional parameter that takes effect only for the nuScenes dataset. It specifies the prefix of the generated info files, with a default value of nuscenes.
--out-dirSpecifies the output directory for preprocessing results.

Usgae Example

Use tools to process the nuScenes dataset.

python3 tools/create_data.py \
    --dataset nuscenes \
    --root-dir /path/to/nuscenes_v1.0-trainval \
    --extra-tag nuscenes_trainval \
    --out-dir ./data/nuscenes_preprocessed