Skip to content

Latest commit

 

History

History
26 lines (17 loc) · 806 Bytes

README.md

File metadata and controls

26 lines (17 loc) · 806 Bytes
Raw

mxnet-SSH

Reproduce SSH (Single Stage Headless Face Detector) with MXNet.

Original Caffe code: https://github.com/mahyarnajibi/SSH

Evaluation on WIDER FACE:

Impelmentation Easy-Set Medium-Set Hard-Set
Original Caffe SSH 0.93123 0.92106 0.84582
Our SSH Model 0.93489 0.92281 0.84525

Installation

  1. Clone the repository.

  2. Download MXNet VGG16 pretrained model from here and put it under model directory.

  3. Type make to build necessary cxx libs.

Training

python train_ssh.py

For more information, check our insightface repository.