pytorch 实现的S3FD,对原来的代码进行了优化,更简洁高效
A PyTorch Implementation of Single Shot Scale-invariant Face Detector
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
python prepare_wider_data.py
We can choose different dataset to train different target[face,head,hand]
python main.py --batch_size 4 --dataset face\hand\head
according to yourself dataset path,modify data/config.py
python afw_test.py
python fddb_test.py
python pascal_test.py
python wider_test.py
you can test yourself image
python demo.py
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