V2X-Seq
收藏Co-MTP 数据集概述
数据集简介
- 名称:Co-MTP
- 描述:Co-MTP 是一个面向自动驾驶的具有多时间融合功能的通用协同轨迹预测框架,利用 V2X 系统完全捕捉历史和未来域中代理之间的交互,以提高规划效果。
- 评估数据集:V2X-Seq
- 性能:Co-MTP 在 V2X-Seq 数据集上取得了 SOTA(当前最佳)性能。
安装环境
- 建议环境:Python 3.8,PyTorch 1.10.1,TensorFlow,DGL 0.7.2,Protobuf 3.20
数据集准备
-
数据下载:下载 V2X-Seq-TFD 数据集并保存到
{DATA_ROOT}目录下。 bash export DATA_ROOT=${PWD}/dataset/V2X-Seq-TFD -
地图合并:将多个地图合并为一个地图。 bash python preprocess/maps_merge.py --data_root ${DATA_ROOT}
-
协同视图轨迹预处理: bash python preprocess/fusion_for_prediction.py --data_root ${DATA_ROOT} --split train python preprocess/preprocess_v2x.py --data_root ${DATA_ROOT} --split train python preprocess/fusion_for_prediction.py --data_root ${DATA_ROOT} --split val python preprocess/preprocess_v2x.py --data_root ${DATA_ROOT} --split val
-
基础设施视图原始轨迹预处理: bash python preprocess/fusion_for_prediction.py --data_root ${DATA_ROOT} --split train python preprocess/preprocess_road.py --data_root ${DATA_ROOT} --split train python preprocess/fusion_for_prediction.py --data_root ${DATA_ROOT} --split val python preprocess/preprocess_road.py --data_root ${DATA_ROOT} --split val
训练与评估
- 训练设备:4 张 NVDIA 4090 显卡
- 训练时间:5-6 天
- 训练命令: bash python train.py --n_epoch 45 --batch_size 24 --val_batch_size 48 --port 49196 --gpu_number 4 --gpu 0,1,2,3 --use_planning True --v2x_prediction True --use_road_obs True --use_other_fut True --road_prediction False --name co-mtp




