Skip to content

This is an implementation in Pytorch of nvidia's model to build a deep learning neural network for self-driving cars.

Notifications You must be signed in to change notification settings

ManajitPal/DeepLearningForSelfDrivingCars

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep Learning For Self-Driving Cars

Goals

The goals / steps of this project are the following:

  • Use the simulator to collect data of good driving behavior.
  • Build, a convolution neural network in Pytorch that predicts steering angles from images.
  • Train and validate the model with a training and validation set.
  • Test that the model successfully drives around the track without leaving the road.

This project is a part of the tutorial published here.

Overview

1. Project contents:

The project includes the following files:

  • self_driving_car.py containing the script to create and train the model.
  • Self_Driving_Car.ipynb containing the notebook code to create and train the model.
  • model.py as a helper function for loading the required model in drive.py.
  • drive.py for driving the car in autonomous mode.
  • model.h5 containing a trained convolution neural network.
  • Note: This project uses python 3 version.

2. Running the model

Using the Udacity provided simulator and the drive.py file, the car can be driven autonomously around the track by executing

python drive.py model.h5

3. Readability

The self_driving_car.py/Self_Driving_Car.ipynb file contains the code for training and saving the convolution neural network. The file shows the pipeline used for training and validating the model, and it contains comments to explain how the code works. The tutorial for the project is here.

Model Architecture

The project is based on the research by Nvidia. The paper is available here. The network is a CNN which maps raw pixel inputs along with steering data from a log file and clones the behaviour of the steering in autonomous mode. For more in depth explanation, please refer to the tutorial.

About

This is an implementation in Pytorch of nvidia's model to build a deep learning neural network for self-driving cars.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published