Awesome
DSFD: Dual Shot Face Detector
A PyTorch Implementation of Dual Shot Face Detector
Description
I use basenet vgg to train DSFD,the model can be downloaded in DSFD.the AP in WIDER FACE as following:
Easy MAP | Medium MAP | hard MAP |
---|---|---|
0.946 | 0.937 | 0.880 |
the AP in AFW,PASCAL,FDDB as following:
AFW | PASCAL | FDDB |
---|---|---|
99.89 | 99.11 | 0.983 |
I'm using resnet50/resnet101 to train DSFD,the result will be published later on
Requirement
- pytorch 0.3
- opencv
- numpy
- easydict
Prepare data
- download WIDER face dataset
- modify data/config.py
python prepare_wider_data.py
Train
python train.py --batch_size 4
--model vgg\resnet50\resnet101
--lr 5e-4
Evalution
according to yourself dataset path,modify data/config.py
- Evaluate on AFW.
python tools/afw_test.py
- Evaluate on FDDB
python tools/fddb_test.py
- Evaluate on PASCAL face
python tools/pascal_test.py
- test on WIDER FACE
python tools/wider_test.py
Demo
you can test yourself image
python demo.py
Result
- demo