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