DAIR-V2X车路协同自动驾驶框架完整使用指南
2026-02-07 04:15:07作者:裴锟轩Denise
项目概述
DAIR-V2X是由清华大学智能产业研究院开发的开源车路协同自动驾驶框架,为研究人员提供完整的数据集、算法实现和评估工具链。该项目包含71,254帧图像和点云数据,支持车辆端与路侧端的协同感知,是自动驾驶领域的重要基础设施。
环境安装与配置
依赖包安装
项目基于mmdetection3d框架开发,需要安装以下核心依赖:
mmdetection3d==0.17.1
pypcd
首先按照官方指南安装mmdetection3d框架,确保版本为0.17.1。对于pypcd包,由于与Python3兼容性问题,需要手动安装修改版本:
git clone https://gitcode.com/gh_mirrors/da/DAIR-V2X
cd DAIR-V2X
python setup.py develop
数据集准备
下载DAIR-V2X-C数据集并按照以下结构组织:
cooperative-vehicle-infrastructure/
├── infrastructure-side/
│ ├── image/{id}.jpg
│ ├── velodyne/{id}.pcd
│ ├── calib/
│ │ ├── camera_intrinsic/{id}.json
│ │ ├── virtuallidar_to_world/{id}.json
│ │ └── virtuallidar_to_camera/{id}.json
│ ├── label/
│ │ ├── camera/{id}.json
│ │ └── virtuallidar/{id}.json
│ └── data_info.json
├── vehicle-side/
│ ├── image/{id}.jpg
│ ├── velodyne/{id}.pcd
│ ├── calib/
│ │ ├── camera_intrinsic/{id}.json
│ │ ├── lidar_to_camera/{id}.json
│ │ ├── lidar_to_novatel/{id}.json
│ │ └── novatel_to_world/{id}.json
│ ├── label/
│ │ ├── camera/{id}.json
│ │ └── lidar/{id}.json
│ └── data_info.json
└── cooperative/
├── label_world/{id}.json
└── data_info.json
创建数据集符号链接:
cd ${dair-v2x_root}/dair-v2x
mkdir ./data/DAIR-V2X
ln -s ${DAIR-V2X-C_DATASET_ROOT}/cooperative-vehicle-infrastructure ./data/DAIR-V2X
系统架构与部署
该架构图展示了完整的车路协同系统部署方案:
- 图(a):路口硬件部署,包含摄像头、激光雷达、路侧单元等设备,实现全方位环境感知
- 图(b):自动驾驶车辆硬件架构,集成多种传感器和计算单元
- 图(c):激光雷达点云数据处理流程
- 图(d):多传感器数据融合结果展示
模型检查点准备
下载预训练模型检查点并放置在指定目录:
configs/vic3d/late-fusion-pointcloud/pointpillars/
├── trainval_config_i.py
├── vic3d_latefusion_inf_pointpillars_596784ad6127866fcfb286301757c949.pth
├── trainval_config_v.py
└── vic3d_latefusion_veh_pointpillars_a70fa05506bf3075583454f58b28177f.pth
模型评估实战
基础评估命令
使用以下命令进行晚期融合点云模型的评估:
cd ${dair_v2x_root}
cd v2x
bash scripts/eval_lidar_late_fusion_pointpillars.sh 0 late_fusion 2 0 100
详细评估配置
或者使用完整的Python命令进行更精细的控制:
rm -r ./cache
cd v2x
DATA="../data/DAIR-V2X/cooperative-vehicle-infrastructure"
OUTPUT="../cache/vic-late-lidar"
rm -r $OUTPUT
rm -r ../cache
mkdir -p $OUTPUT/result
mkdir -p $OUTPUT/inf/lidar
mkdir -p $OUTPUT/veh/lidar
python eval.py \
--input $DATA \
--output $OUTPUT \
--model late_fusion \
--dataset vic-async \
--k 2 \
--split val \
--split-data-path ../data/split_datas/cooperative-split-data.json \
--inf-config-path ../configs/vic3d/late-fusion-pointcloud/pointpillars/trainval_config_i.py \
--inf-model-path ../configs/vic3d/late-fusion-pointcloud/pointpillars/vic3d_latefusion_inf_pointpillars_596784ad6127866fcfb286301757c949.pth \
--veh-config-path ../configs/vic3d/late-fusion-pointcloud/pointpillars/trainval_config_v.py \
--veh-model-path ../configs/vic3d/late-fusion-pointcloud/pointpillars/vic3d_latefusion_veh_pointpillars_a70fa05506bf3075583454f58b28177f.pth \
--device 0 \
--pred-class car \
--sensortype lidar \
--extended-range 0 -39.68 -3 100 39.68 1
关键参数说明
- input:数据集根目录路径
- output:预测结果输出目录
- model:模型类型,可选veh_only、inf_only、late_fusion、early_fusion
- dataset:数据集名称,可选dair-v2x-v、dair-v2x-i、vic-sync、vic-async
- k:用于vic-async数据集的前帧数量
- pred_class:评估的预测类别
- sensortype:输入数据类型,可选lidar、camera、multimodality
- extended-range:车辆中心坐标系下的感兴趣区域范围
基准结果复现
项目提供了多种模态和融合策略的预训练模型检查点:
| 模态 | 融合策略 | 模型 | 数据集 | 详细说明 |
|---|---|---|---|---|
| Image | VehOnly | ImvoxelNet | VIC-Sync | README |
| Image | InfOnly | ImvoxelNet | VIC-Sync | README |
| Pointcloud | VehOnly | PointPillars | VIC-Sync | README |
| Pointcloud | InfOnly | PointPillars | VIC-Sync | README |
| Pointcloud | Late-Fusion | PointPillars | VIC-Sync | README |
核心功能模块使用
数据集加载
使用项目提供的数据加载器轻松加载DAIR-V2X数据集,具体方法参考:数据加载器文档
可视化功能
支持LiDAR和摄像头帧的可视化,详细操作指南:可视化文档
自定义模型评估
框架支持评估用户自定义的VIC3D目标检测模型,参考:自定义模型文档
融合模块应用
深入了解和使用融合模块,请参考:融合模块文档
实用技巧与最佳实践
- 内存优化:合理配置GPU内存使用,避免内存溢出
- 数据预处理:利用分布式数据加载加速训练过程
- 模型选择:根据具体场景选择合适的融合策略和模型架构
- 性能监控:实时监控训练进度和模型性能指标
技术支持与资源
- 官方文档:docs/get_started.md
- 配置示例:configs/vic3d/
- 数据处理工具:tools/dataset_converter/
DAIR-V2X框架为车路协同自动驾驶研究提供了强大的技术支撑,帮助研究人员快速开展算法开发和实验验证。
登录后查看全文
热门项目推荐
相关项目推荐
GLM-5智谱 AI 正式发布 GLM-5,旨在应对复杂系统工程和长时域智能体任务。Jinja00
GLM-5.1GLM-5.1是智谱迄今最智能的旗舰模型,也是目前全球最强的开源模型。GLM-5.1大大提高了代码能力,在完成长程任务方面提升尤为显著。和此前分钟级交互的模型不同,它能够在一次任务中独立、持续工作超过8小时,期间自主规划、执行、自我进化,最终交付完整的工程级成果。Jinja00
MiniMax-M2.7MiniMax-M2.7 是我们首个深度参与自身进化过程的模型。M2.7 具备构建复杂智能体应用框架的能力,能够借助智能体团队、复杂技能以及动态工具搜索,完成高度精细的生产力任务。Python00- QQwen3.5-397B-A17BQwen3.5 实现了重大飞跃,整合了多模态学习、架构效率、强化学习规模以及全球可访问性等方面的突破性进展,旨在为开发者和企业赋予前所未有的能力与效率。Jinja00
HY-Embodied-0.5这是一套专为现实世界具身智能打造的基础模型。该系列模型采用创新的混合Transformer(Mixture-of-Transformers, MoT) 架构,通过潜在令牌实现模态特异性计算,显著提升了细粒度感知能力。Jinja00
LongCat-AudioDiT-1BLongCat-AudioDiT 是一款基于扩散模型的文本转语音(TTS)模型,代表了当前该领域的最高水平(SOTA),它直接在波形潜空间中进行操作。00
项目优选
收起
deepin linux kernel
C
27
14
OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
659
4.26 K
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
1.54 K
894
Ascend Extension for PyTorch
Python
504
609
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
391
288
暂无简介
Dart
906
218
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
69
21
昇腾LLM分布式训练框架
Python
142
168
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
939
863
🍒 Cherry Studio 是一款支持多个 LLM 提供商的桌面客户端
TypeScript
1.33 K
108
