ACT-Bench
收藏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}, }




