Skip to content
forked from rykov8/ssd_keras

Port of Single Shot MultiBox Detector to Keras

License

Notifications You must be signed in to change notification settings

KaiserW/ssd_keras

 
 

Repository files navigation

For more details, please refer to arXiv paper.

SSD architecture

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 v2.1.6, Tensorflow v1.7.0, OpenCV v3.1.4

About

Port of Single Shot MultiBox Detector to Keras

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 89.6%
  • Python 10.4%