/SSD-Object-Detection

SSD Object Detection

Primary LanguagePythonThe UnlicenseUnlicense

SSD-Object-Detection

In this project, I have used SSD512 algorithm to detect objects in images and videos.
This project use prebuild model and weights.

For Original Model creation and training on your own datasets, please check out Pierluigi Ferrari' SSD Implementation

Overview

SSD (Single Shot MultiBox Detector)

  • Single Shot: this means that the tasks of object localization and classification are done in a single forward pass of the network
  • MultiBox: this is the name of a technique for bounding box regression developed by Szegedy et al.
  • Detector: The network is an object detector that also classifies those detected objects

Examples

Test Image
Test Image
Test Image

Dependencies

  • Python 3+
  • Keras (with tensorflow backend)
  • OpenCV
  • Numpy
  • imageio
  • ffmpeg

How to use

  1. Fork this repo
  2. Download the pretrained weights by the following link and save it in weights folder. Weigths
  3. Put the test images under inputs/images and videos under inputs/videos
  4. Execute the SSD-Object-Detection.py file.
  5. Voilà, check the outputs folder.

References