GitHub - yxlijun/S3FD.pytorch: SFD implement with pytorch (original) (raw)

S³FD: Single Shot Scale-invariant Face Detector

A PyTorch Implementation of Single Shot Scale-invariant Face Detector

Description

Meanwhile train hand and head with S3FD,hand dataset is Egohands Dataset,head dataset is SCUT-HEAD,we can download hand model and face model

Requirement

Prepare data

  1. download WIDER face dataset、Egohands dataset and SCUT-HEAD
  2. modify data/config.py
  3. python prepare_wider_data.py4 python prepare_handataset.py

Train

We can choose different dataset to train different target[face,head,hand]

python train.py --batch_size 4 --dataset face\hand\head

Evalution

according to yourself dataset path,modify data/config.py

  1. Evaluate on AFW.
  2. Evaluate on FDDB
  3. Evaluate on PASCAL face
  4. test on WIDER FACE

Demo

you can test yourself image

Result

  1. AFW PASCAL FDDB

afw pascal fddb 

AFW AP=99.81 paper=99.85 
PASCAL AP=98.77 paper=98.49
FDDB AP=0.975 paper=0.983
WIDER FACE:
Easy AP=0.925 paper = 0.927
Medium AP=0.925 paper = 0.924
Hard AP=0.854 paper = 0.852
  1. demo

afw

References