NGSIM+INTERACTION datasets
收藏ASG(LC): Adversarial Scenario Generation for Lane-change 数据集概述
数据集简介
- 目标:从真实轨迹数据(NGSIM+INTERACTION数据集)中提取典型的关键变道驾驶场景,并生成新的对抗性变道场景。
- 生成方法:使用GAIL(生成对抗模仿学习),包含PPO改进方法、奖励函数的创新设计、策略网络的优化。
数据集结构
-
数据集文件夹结构:
datasets |-- Interaction | |-- INTERACTION-Dataset-TC-v1_0 | | |-- maps | |
-- recorded_trackfiles | |-- DR_USA_Intersection_MA | | |-- vehicle_tracks_000.csv | | |-- ... | |-- vehicle_tracks_021.csv | |-- maps | |-- recorded_trackfiles | | |-- DR_CHN_Merging_ZS | | |-- ... | |-- TC_BGR_Intersection_VA | |-- INTERACTION-Dataset-DR-v1_1.rar |-- INTERACTION-Dataset-TC-v1_0.zip-- ngsim.csv
数据处理
1.1 数据集准备
- 将NGSIM数据集与INTERACTION数据集链接或复制到
data/datasets文件夹。
1.2 场景提取
-
高速公路场景: shell cd NGSIM_env/data python3 data_process.py
-
城市交叉口场景: shell cd NGSIM_env/data python3 interaction.py
1.3 专家轨迹提取
-
高速公路场景: shell cd examples/data_process python3 naturalistic_ngsim.py
-
城市交叉口场景: shell cd examples/data_process python3 naturalistic_interaction.py
自然交通流生成
2.1 人类驾驶策略建模
-
高速公路场景: shell cd examples/av2_model_train python3 train.py --policy human_driving_policy --scenario highway --model diffusion --num-epochs 1000 --num-threads 6 --exp_name exp_1 [--resume]
-
城市交叉口场景: shell cd examples/av2_model_train python3 train.py --policy human_driving_policy --scenario intersection --model diffusion --num-epochs 1000 --num-threads 6 --exp_name exp_1 [--resume]
2.2 自然交通流生成
-
高速公路场景: shell cd examples/av2_model_test python3 test.py --policy human_driving_policy --scenario highway --model diffusion --load-model-id 400 --num-threads 6 --exp_name exp_1 --epochs 1000 [--show] [--save-video]
-
城市交叉口场景: shell cd examples/av2_model_test python3 test.py --policy human_driving_policy --scenario intersection --model diffusion --load-model-id 400 --num-threads 6 --exp_name exp_1 --epochs 1000 [--show] [--save-video]
自然对抗测试场景生成
3.1 自然对抗策略建模
-
高速公路场景: shell cd examples/av2_model_train python3 train.py --policy natural_adversarial_model --scenario highway --model ppo --num-epochs 1000 --num-threads 6 --exp_name exp_1 --supervise-model diffusion --supervise-model-id 3700 [--resume]
-
城市交叉口场景: shell cd examples/av2_model_train python3 train.py --policy natural_adversarial_model --scenario intersection --model ppo --num-epochs 1000 --num-threads 6 --exp_name exp_1 --supervise-model diffusion --supervise-model-id 4899 [--resume]
3.2 自然对抗测试场景生成
-
高速公路场景: shell cd examples/av2_model_test python3 train.py --policy natural_adversarial_model --scenario highway --model ppo --load-model-id 400 --num-threads 6 --exp_name exp_1 --supervise-model diffusion --supervise-model-id 3700 [--show] [--save-video]
-
城市交叉口场景: shell cd examples/av2_model_test python3 train.py --policy natural_adversarial_model --scenario intersection --model ppo --load-model-id 400 --num-threads 6 --exp_name exp_1 --supervise-model diffusion --supervise-model-id 3700 [--show] [--save-video]




