ACT-Bench
收藏资源简介:
ACT-Bench是一个专门用于定量评估自动驾驶世界模型动作可控性的框架。它专注于测量世界模型在指定轨迹条件下生成驾驶场景的能力。
ACT-Bench is a framework dedicated to the quantitative evaluation of action controllability for autonomous driving world models. It focuses on measuring the capability of such models to generate driving scenarios under specified trajectory conditions.
ACT-Bench 数据集概述
数据集简介
ACT-Bench 是一个专门用于定量评估自动驾驶世界模型动作可控性的框架。其核心目标是测量世界模型在给定轨迹条件下生成驾驶场景的能力。
数据集内容
- 数据集样本数量: 2286
- 数据集格式: 视频文件(MP4)
- 数据集存储位置:
generated_videos/<your_model_name>/NUSCENES_ACTION_*.mp4
数据集使用
1. 生成视频
- 步骤: 使用世界模型在基准数据集上生成视频。
- 注意事项:
instruction_trajs可能需要根据特定世界模型的输入格式进行调整。 - 示例: 参考 Terra/generate.py 中的
prepare_action()函数。
2. 计算评分
-
工具: ACT-Estimator
-
命令: bash python run_benchmark.py --input_dir generated_videos/<your_model_name> --output_dir results/<your_model_name>
-
脚本示例: python from act_bench import ActBenchConfig, compute_score
config = ActBenchConfig( input_dir="generated_videos/<your_model_name>", output_dir="results/<your_model_name>", ) results = compute_score(config) print(f"Accuracy: {results.accuracy*100:.2f}%") print(f"Mean ADE: {results.ade:.4f}, Mean FDE: {results.fde:.4f}")
论文结果复现
-
下载生成视频: bash python download_generated_videos.py
-
计算评分: bash
For Terra
./scripts/compute_score_terra_paper.sh
For Vista
./scripts/compute_score_vista_paper.sh
For Terra v2
./scripts/compute_score_terra_v2.sh
-
Notebook 示例: compute_score.ipynb
引用
如果论文或代码对您的研究有帮助,请考虑引用: bibtex @misc{arai2024actbench, title={ACT-Bench: Towards Action Controllable World Models for Autonomous Driving}, author={Hidehisa Arai and Keishi Ishihara and Tsubasa Takahashi and Yu Yamaguchi}, year={2024}, eprint={2412.05337}, archivePrefix={arXiv}, primaryClass={cs.CV}, url={https://arxiv.org/abs/2412.05337}, }




