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
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
- Python 3+
- Keras (with tensorflow backend)
- OpenCV
- Numpy
- imageio
- ffmpeg
- Fork this repo
- Download the pretrained weights by the following link and save it in weights folder.
Weigths
- Put the test images under
inputs/images
and videos underinputs/videos
- Execute the
SSD-Object-Detection.py
file. - Voilà, check the
outputs
folder.
- Original Keras implementation of SSD by Pierluigi Ferrari
- Orginal Paper SSD: Single Shot MultiBox Detector