Skip to content

[T-ITS] Driving Behavior Modeling using Naturalistic Human Driving Data with Inverse Reinforcement Learning

License

Notifications You must be signed in to change notification settings

MCZhi/Driving-IRL-NGSIM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

35 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Driving-IRL-NGSIM

This repo is the implementation of the paper "Driving Behavior Modeling using Naturalistic Human Driving Data with Inverse Reinforcement Learning". It contains NGSIM env that can replay vehicle trajectories in the NGSIM dataset while also simulate some interactive behaviors, as well as inverse reinforcement learning (IRL) implementation in this paper for learning driver's reward function.

Driving Behavior Modeling using Naturalistic Human Driving Data with Inverse Reinforcement Learning
Zhiyu Huang, Jingda Wu, Chen Lv
AutoMan Research Lab, Nanyang Technological University
[Paper] [arXiv]

Getting started

Install the dependent package

pip install -r requirements.txt

Download the NGSIM dataset from this website (export to csv file) and run dump_data.py along with the path to the downloaded csv file (may take a while)

python dump_data.py [YOUR PATH]/Next_Generation_Simulation__NGSIM__Vehicle_Trajectories_and_Supporting_Data.csv

Run IRL personalized or IRL general

python personal_IRL.py 
python general_IRL.py 

Reference

If you find this repo to be useful in your research, please consider citing our work

@article{huang2021driving,
  title={Driving Behavior Modeling Using Naturalistic Human Driving Data With Inverse Reinforcement Learning},
  author={Huang, Zhiyu and Wu, Jingda and Lv, Chen},
  journal={IEEE Transactions on Intelligent Transportation Systems},
  year={2021},
  publisher={IEEE}
}

License

This repo is released under the MIT License. The NGSIM data processing code is borrowed from NGSIM interface. The NGSIM env is built on top of highway env which is released under the MIT license.

About

[T-ITS] Driving Behavior Modeling using Naturalistic Human Driving Data with Inverse Reinforcement Learning

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages