Port of Single Shot MultiBox Detector to Keras
Switch branches/tags
Nothing to show
Clone or download
Pull request Compare This branch is 1 commit ahead of rykov8:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
PASCAL_VOC
pics
testing_utils
.gitignore
LICENSE
README.md
SSD.ipynb
SSD_training.ipynb
gt_pascal.pkl
prior_boxes_ssd300.pkl
ssd.py
ssd_layers.py
ssd_training.py
ssd_utils.py

README.md

license

A port of SSD: Single Shot MultiBox Detector to Keras framework.

For more details, please refer to arXiv paper. For forward pass for 300x300 model, please, follow SSD.ipynb for examples. For training procedure for 300x300 model, please, follow SSD_training.ipynb for examples. Moreover, in testing_utils folder there is a useful script to test SSD on video or on camera input.

Weights are ported from the original models and are available here. You need weights_SSD300.hdf5, weights_300x300_old.hdf5 is for the old version of architecture with 3x3 convolution for pool6.

This code was tested with Keras v1.2.2, Tensorflow v1.0.0, OpenCV v3.1.0-dev