Skip to content

Autonomous-Robotics-UTM/Imitation-Learning-Net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Imitation Learning Neural Network

The network takes in images and labels of manually driven vehicles and learns to predict actions for new images.

Setup Process

Navigate to /net and create a directory called /data. Insert the Dataset after unzipping it.

Run Trainer.ipynb to train the neural network. Saved model versions will be saved in the snapshots folder

In order to monitor the loss you will have to install tensorboard with the following command: pip install tensorboard

In order to see the update in the loss you can view the results on Tensorboard by:

  1. Opening a new terminal window
  2. run 'tensorboard --logdir runs' in the terminal
  3. Navigating to the link displayed on the terminal

Important

While we don't have gitignore set up, don't push commits with the dataset in the file structure, there's no point of having the dataset backed up on github.

Online Report

Latex doc

PDF

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •