This repository contains the code and related materials for the project on motion prediction of road users at intersections using a Long Short-Term Memory (LSTM)-based neural network.
With the increasing demand for intelligent autonomous systems on roads and in human environments, the capacity of such systems to sense, analyze, and predict behavior is highly important. This project focuses on the motion prediction of road users, particularly at intersections, using a neural network-based approach, specifically LSTM. The project provides a multivariate, multi-output, multi-step LSTM model to forecast the future positions of dynamic agents.
The project directory has the following structure:
data_processing: Contains scripts or notebooks for data preprocessing.evaluation: Includes scripts or notebooks for evaluating the performance of the prediction models.pictures_main: Contains any visual materials or images related to the main project.prediction_models: Consists of the main scripts or notebooks for training and testing the LSTM prediction models.visualization: Includes scripts or notebooks for visualizing the results..DS_Store: macOS system file storing custom attributes of a folder.main.ipynb: The main Jupyter notebook containing the overall project implementation.NN_prediction_result_Model_1_TestingID_28.xlsx: Excel file containing prediction results for Model 1 on testing data.NN_prediction_result_Model_2_TestingID_28.xlsx: Excel file containing prediction results for Model 2 on testing data.Previously_trained_Model_1.h5: Previously trained weights for Model 1.Previously_trained_Model_2.h5: Previously trained weights for Model 2.
To replicate or extend the project, follow these steps:
-
Clone the repository:
git clone https://github.com/[your-username]/motion-prediction.git
The project involves several components:
Data Processing: Check the data_processing directory for scripts or notebooks related to data preprocessing.
Prediction Models: The core of the project is in the prediction_models directory, where LSTM models are trained and tested for motion prediction.
Evaluation: Evaluate the performance of the models using scripts or notebooks in the evaluation directory.
Visualization: Visualize the results with scripts or notebooks found in the visualization directory.
Results: Review the prediction results in the Excel files (NN_prediction_result_Model_1_TestingID_28.xlsx and NN_prediction_result_Model_2_TestingID_28.xlsx).