Skip to content

Use 3D convolutional network to predict the speed of a car via dashcam stream. The final mean squared error on validation set is 3.1 (mile/hour)^2.

Notifications You must be signed in to change notification settings

experiencor/speed-prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is a proposal to solve the speed prediction challenge posted by https://comma.ai/. 3D Convolutional Network [1] is used to predict car speed using dashcam footage. The trained network is good enough to make reasonable prediction on arbitrary footages.

The final mean squared error on validation set is 3.1!

The result on comma.ai validation footage (MSE of 3.1):

Foo

The result on 28 minute udacity footage (MSE of 23.28):

Foo

Note that this is the result when the network is purely trained on comma.ai data and tested on udacity data with completely different lighting conditions and different scene dynamics. Most of the error on the udacity footage occurs at the turns, which is limitedly seen in comma.ai dataset.

[1] Learning Spatiotemporal Features with 3D Convolutional Networks (https://arxiv.org/abs/1412.0767)

[2] Unsupervised Learning of Video Representations using LSTMs (https://arxiv.org/abs/1502.04681)

Take a look at a similar attempt at https://github.com/JonathanCMitchell/speedChallenge

Information

Network architecture

Validation losses

Usage

  1. Download the weights pretrained on the Sports-1M dataset [https://goo.gl/tsEUo2]

  2. Download the training data from comma.ai website [https://goo.gl/ERi7Uh]

  3. Follow the steps in "Dashcam Speed - C3D.ipynb" for a walk-through of the training and testing of the network

Todo list

  1. Use shorter look-back time window to allow more fine-grained speed prediction. At the moment, the network looks 16 frames back in time and predicts the average speed during this period.
  2. Use semantic segmentation to cut off moving objects. Moving objects such as vehicles confuses the speed prediction w.r.t. the road. More data with diverse conditions may also be able to clear this issue.

About

Use 3D convolutional network to predict the speed of a car via dashcam stream. The final mean squared error on validation set is 3.1 (mile/hour)^2.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published