This is the code repository for our paper, Driving Behavior Classification Based on Sensor Data Fusion Using LSTM Recurrent Neural Networks.
This repo contains the data and tensorflow code for training and testing the LSTM model we used in the paper. To run this code, it is necessary to have installed:
- Python2.7
- Tensorflow == 0.12
- Numpy
- Sklearn
If you find this work useful in your research, please consider citing:
@inproceedings{saleh2017driving,
title={Driving behavior classification based on sensor data fusion using lstm recurrent neural networks},
author={Saleh, Khaled and Hossny, Mohammed and Nahavandi, Saeid},
booktitle={2017 IEEE 20th International Conference on Intelligent Transportation Systems (ITSC)},
pages={1--6},
year={2017},
organization={IEEE}
}
- Clone repo via
git clone https://github.com/KhaledSaleh/driving_behaviour_classification.git && cd driving_behaviour_classification
. - Run
python main.py --help
to check the available command line args. - Run
python main.py
to train the model on the full dataset.