/self-driving

Machine learning models for self-driving cars

Primary LanguagePythonMIT LicenseMIT

self-driving

Machine learning models for self-driving cars. This repo is a WIP and will be updated regularly.

Requires:

Models and Tasks

Prediction with deep learning

The specific task here is to predict driving speed from a dashboard-mounted camera; there's tons of this data open-sourced by the folks at comma.ai here. I experimented with several flavors of deep nets:

  1. Static CNN -- The model is based off of NVIDIA's "End-to-end Self-driving" model. It's static because each video frame is assumed independent by the model, ignoring temporal dependencies. The resulting model is in models/cnn_prediction.py.
  2. CNN + LSTM -- In order to incorporate temporal info into the static CNN, I experimented with several architectures discussed in "Beyond Short Snippets: Deep Networks for Video Classification". The resulting model is in models/cnn_lstm_prediction.py.
  3. Optical flow -- Tracking feature points with sparse optical flow implicitly encodes temporal dependencies between frames. (This model will be pushed soon!)

Check out speed_prediction.md for more.

![](training_screenshot.png?raw=true "Screenshot from training" =200x)

Datasets

A few places to find open-source data to play with: