数据集处理工具

工具链提供tools/create_data.py工具用于 KITTI-3D 和 nuScenes 两个数据集的预处理。

使用方法

运行tools/create_data.py,即可对目标数据集进行预处理。

参数介绍

参数描述
--dataset数据集名称,可选值为 KITTI-3D 和 nuScenes
--root-dir数据集所在根目录。
--extra-tag可选参数,仅 nuScenes 数据集生效,指定生成信息文件的前缀,默认值为nuscenes。
--out-dir指定预处理结果输出目录

使用示例

使用工具对 nuScenes 数据集进行处理。

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