Lane change prediction based on LSTM using HighD dataset.
lane-change-prediction-lstm/ .... Top src dir
|-- LICENSE ..................... Full license text
|-- main.py ..................... main file executing data processing module
|-- process_data.py ............. data processing library
|-- read_data.py ................ read data functionality for data processing
|-- rnn_model.py ................ LSTM training and testing
|-- requirement.txt ............. Requirements for building
-
Please put the HighD dataset
data/in the directory. -
Run
python3 main.pyto process the dataset and the data will be stored intooutput/in pickle format. -
Run
python3 rnn_modelto perform the training and testing.
Packages installation guide: pip3 install -r requirement.txt
Please use Python3
All rights reserved.