Original Caffe code: https://github.com/mahyarnajibi/SSH
Deepinsight SSH Mxnet code: https://github.com/deepinsight/mxnet-SSH
Evaluation on WIDER FACE:
| Impelmentation | Easy-Set | Medium-Set | Hard-Set |
|---|---|---|---|
| Original Caffe SSH | 0.93123 | 0.92106 | 0.84582 |
| Deepinsight SSH Model | 0.93489 | 0.92281 | 0.84525 |
| Our SSHA Model | - | - | - |
-
Clone the repository.
-
Download MXNet VGG16 pretrained model from here and put it under
modeldirectory. -
Type
maketo build necessary cxx libs.
mkdir datacd dataln -s $(widerface dir) widerface- You should make keypoint label 'wider_face_train_bbx_kp_gt.txt' instead of 'wider_face_train_bbox_gt.txt'
python train_ssh.py