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 |
-
Clone the repository.
-
Download MXNet VGG16 pretrained model from here and put it under
model
directory. -
Type
make
to build necessary cxx libs.
python train_ssh.py
For more information, check our insightface repository.