Vehicle Detection system

How I built a vehicle detection system for self driving cars

Project description

In this project, I implemented a pipeline to detect vehicles in images. The following steps were performed:

  1. Extract features using Histogram of Oriented Gradients (HOG) on a labeled training set of images
  2. Train a Linear Support Vector Machine classifier
  3. Implement a sliding-window technique and use the trained classifier to search for vehicles in images.
  4. Run the pipeline on a video stream and create a heat map of recurring detections frame by frame to reject outliers and follow detected vehicles.
  5. Estimate a bounding box for vehicles detected.

The end result is shown below, with the full code available in the jupyter notebook.

Training data

This project used labeled data for vehicle and non-vehicle examples to train the classifier. These example images come from a combination of the GTI vehicle image database, the KITTI vision benchmark suite, and examples extracted from the project video itself.

Technologies Used

python
jupyter
opencv
numpy
scikit-learn