mxnet-SSH
mxnet-SSH copied to clipboard
Reproduce SSH (Single Stage Headless Face Detector) with MXNet
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
-
Clone the repository.
-
Download MXNet VGG16 pretrained model from here and put it under
modeldirectory. -
Type
maketo build necessary cxx libs.
Training
python train_ssh.py
For more information, check our insightface repository.