SemanticTHAB
收藏数据集概述
数据集简介
- 名称: Real Time Semantic Segmentation of High Resolution Automotive LiDAR Scans
- 用途: 用于训练和微调Ouster OS2-128 (Rev 7) LiDAR数据的语义分割模型
- 数据来源: TH AB
包含的数据集
-
SemanticKITTI
- 规模: 22,000+ 3D LiDAR点云
- 环境: 城市环境
- 标注: 28个语义类别(如道路、汽车、行人、建筑物等)
- 用途: 自动驾驶中的语义分割算法训练与评估
-
SemanticTHAB
- 规模: 4,000+ 3D LiDAR点云
- 环境: 城市环境
- 标注: 与SemanticKITTI共享标签定义
- 下载地址: https://zenodo.org/records/14906179
- DOI: 10.5281/zenodo.14906179
开发环境
- 操作系统: Ubuntu 22.04.4 LTS x86_64
- 硬件配置:
- CPU: AMD Ryzen 9 3900X (24) @ 3.800G
- GPU: NVIDIA GeForce RTX 3090
- 内存: 32031MiB
- 开发工具: VS-Code,使用远程容器开发
训练与推理
-
训练SemanticKITTI
-
单次训练命令:
python train_semantic_Kitti.py --model_type resnet34 --learning_rate 0.001 --num_epochs 50 --batch_size 1 --num_workers 1 --rotate --flip --visualization -
批量训练脚本:
./run_training_kitti.sh -
输出结构:
dataset └───train_semantic_KITTI └───{backbone}_{model_configuration} │ config.json │ events.out.tfevents.* │ model_final.pth │ results.json
-
-
训练SemanticTHAB
-
训练命令:
python repos/train_semantic_THAB.py --model_type resnet34 --learning_rate 0.001 --num_epochs 50 --batch_size 8 --num_workers 16 --rotate --flip --visualization -
批量训练脚本:
./run_training_THAB.sh -
输出结构:
dataset └───train_semantic_THAB └───{backbone}_{model_configuration} │ config.json │ events.out.tfevents.* │ model_final.pth │ results.json
-
-
推理示例
- 示例脚本:
/src/inference_ouster.py
- 示例脚本:
ROS演示系统
许可证
- 项目许可证: Apache 2.0 License
- 数据集许可证: 单独提供,需另行查看

- 1Real Time Semantic Segmentation of High Resolution Automotive LiDAR Scans应用科学阿沙芬堡大学工程系, 卡塞尔大学智能嵌入式系统实验室 · 2025年



