Skip to content

Project for CS236G involving transformers and GANs. Predict drone trajectory.

Notifications You must be signed in to change notification settings

aksbaih/drone-trajectory

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

67 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

drone-trajectory

Project for Stanford CS236G involving transformers and GANs. Refer to the report for more.

The task at hand is to predict multiple possible trajectories of a drone for the next second given observations of its location for the past 1.5 seconds.

  • ./baseline/ shows the implementation and instructions to train and run the baseline transformer.
  • ./gan/ shows the implementation and instructions to train and run my new 3 approaches for a GAN-Transformer-based model to predict a variety of trajectories.

Submodules

This repo contains the following submodule repos:

To clone correctly, use the following command

git clone --recurse-submodules https://github.com/aksbaih/drone-trajectory

Future Thoughts

GAN's are not the best approach for this task as there are different architectures such as GPT which proved good performance in similar tasks in NLP and require less training.

About

Project for CS236G involving transformers and GANs. Predict drone trajectory.

Resources

Stars

Watchers

Forks