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
Also support newest Keras v2.0.1 (using ssd_v2.py)

Name		Name	Last commit message	Last commit date
Latest commit History 44 Commits
PASCAL_VOC		PASCAL_VOC
pics		pics
testing_utils		testing_utils
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
SSD.ipynb		SSD.ipynb
SSD_training.ipynb		SSD_training.ipynb
feature_flow.py		feature_flow.py
flow_util.py		flow_util.py
gt_pascal.pkl		gt_pascal.pkl
plot_util.py		plot_util.py
prior_boxes_ssd300.pkl		prior_boxes_ssd300.pkl
run_ssd.py		run_ssd.py
ssd.py		ssd.py
ssd_conv4_3.py		ssd_conv4_3.py
ssd_layers.py		ssd_layers.py
ssd_training.py		ssd_training.py
ssd_utils.py		ssd_utils.py
ssd_v2.py		ssd_v2.py
test_keras.py		test_keras.py
test_ssd.py		test_ssd.py

License

cory8249/ssd_keras

Folders and files

Latest commit

History

Repository files navigation

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

About

Resources

License

Stars

Watchers

Forks

Languages