Alert button

Trajectory Prediction using Generative Adversarial Network in Multi-Class Scenarios

Oct 18, 2021
Shilun Li, Tracy Cai, Jiayi Li

Figure 1 for Trajectory Prediction using Generative Adversarial Network in Multi-Class Scenarios
Figure 2 for Trajectory Prediction using Generative Adversarial Network in Multi-Class Scenarios
Figure 3 for Trajectory Prediction using Generative Adversarial Network in Multi-Class Scenarios
Figure 4 for Trajectory Prediction using Generative Adversarial Network in Multi-Class Scenarios

Share this with someone who'll enjoy it:

Predicting traffic agents' trajectories is an important task for auto-piloting. Most previous work on trajectory prediction only considers a single class of road agents. We use a sequence-to-sequence model to predict future paths from observed paths and we incorporate class information into the model by concatenating extracted label representations with traditional location inputs. We experiment with both LSTM and transformer encoders and we use generative adversarial network as introduced in Social GAN to learn the multi-modal behavior of traffic agents. We train our model on Stanford Drone dataset which includes 6 classes of road agents and evaluate the impact of different model components on the prediction performance in multi-class scenes.

View paper onarxiv icon

Share this with someone who'll enjoy it: