Skip to content

sharmasushil/Navigating-Uncertainty-Trajectory-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 

Repository files navigation

Navigating Uncertainty: The Role of Short-Term Trajectory Prediction in Autonomous Vehicle Safety 🚘

Abstract: Autonomous vehicles require accurate and reliable short-term trajectory predictions for safe and efficient driving. While most commercial automated vehicles currently use state machine-based algorithms for trajectory forecasting, recent efforts have focused on end-to-end data-driven systems. Often, the design of these models is limited by the availability of datasets, which are typically restricted to generic scenarios. To address this limitation, we have developed a synthetic dataset for short-term trajectory prediction tasks using the CARLA simulator. This dataset is extensive and incorporates what is considered complex scenarios - pedestrians crossing the road, vehicles overtaking - and comprises 6000 perspective view images with corresponding IMU and odometry information for each frame. Furthermore, an end-to-end short-term trajectory prediction model using convolutional neural networks (CNN) and long short-term memory (LSTM) networks has also been developed. This model can handle corner cases, such as slowing down near zebra crossings and stopping when pedestrians cross the road, without the need for explicit encoding of the surrounding environment. To accelerate this research and assist others, we are releasing our dataset and model to the research community.

Perspective View Images from Carla 📌

This repository contains a collection of perspective-view images captured from the Carla simulator. The images are intended for research and development purposes in the fields of computer vision, machine learning, and autonomous driving.

Architecture Topology ⛓️

The network developed in this work is designed to predict the future trajectories of a vehicle based on a sequence of perspective-view images. It comprises two primary components: a Convolutional Neural Network (CNN) and a Long-Short Term Memory Network (LSTM

The following diagram illustrates a suggested topology for a CNN architecture designed to estimate trajectories from image sequences. It highlights the progression of information flow across convolutional and fully connected layers.

Dataset Description: 💻

The dataset comprises a set of perspective-view images captured from different camera angles and positions within the Carla simulator environment. The images are saved in PNG format and have a resolution of 800 x 600 pixels. Each image is associated with a timestamp indicating the time of capture.

Ablation Study 📖

An ablation study on the number of LSTM cells ($\alpha$=1, $\beta$=2, $\gamma$=3, $\delta$=4) is conducted on our CNN-LSTM model. This comparison was performed using the CARLA dataset for the two specified levels, Level 1 and Level 2 respectively. For this analysis, three evaluation metrics - ARMSE, AMAPE, and AED.

  • ARMSE: Average Root Mean Square Error

  • AMAPE: Average Mean Absolute Percentage Error

  • AED: Average Euclidean Distance

CARLA: Dataset Model ARMSE ↓ AMAPE ↓ AED ↓
Dataset: Level 1 CNN-LSTM ($\alpha$)
CNN-LSTM ($\beta$)
CNN-LSTM ($\gamma$)
CNN-LSTM ($\delta$)
0.0046
0.0034
0.0028
0.0024
0.0056
0.0043
0.0038
0.0033
0.0050
0.0039
0.0032
0.0028
Dataset: Level 2 CNN-LSTM ($\alpha$)
CNN-LSTM ($\beta$)
CNN-LSTM ($\gamma$)
CNN-LSTM ($\delta$)
0.0126
0.0097
0.0082
0.0065
0.0172
0.0133
0.0119
0.0107
0.0154
0.0127
0.0107
0.0079

Table 1: Analysis of ARMSE, AMAPE, and AED

Usage ⚙️

The images in this dataset can be used for various purposes, including:

  • Training and evaluating computer vision algorithms
  • Developing and testing machine learning models for trajectory prediction, object detection, semantic segmentation, or other vision tasks
  • Conducting research in the field of autonomous driving and perception systems

Citation👇🏻

If you use this dataset in your research or work, we kindly request that you cite the following paper:

@inproceedings{sharma2023navigating,
      title={Navigating Uncertainty: The Role of Short-Term Trajectory Prediction in Autonomous Vehicle Safety}, 
      author={Sushil Sharma and Ganesh Sistu and Lucie Yahiaoui and Arindam Das and Mark Halton and Ciarán Eising},
      year={2023},
      booktitle={Proceedings of the Irish Machine Vision and Image Processing Conference}
}

Download Dataset 📂

To access the dataset, please click Dataset Download (< 1 GB) to visit the download page.

Data statistics for each class of our newly created dataset

CARLA Results Visualization 📈

We showcase the resilience of our model in dealing with challenging scenarios, including pedestrian crossings, without relying on the explicit encoding of contextual information. The figure below illustrates a comparison between Ground Truth (shown in blue) and Predicted Bounding Boxes (highlighted in red) for trajectory prediction.

Contact ✉️

For any questions or inquiries regarding the dataset, please contact [email protected]

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages