Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/yxlijun/Pyramidbox.pytorch
Pyramidbox implement with pytorch
https://github.com/yxlijun/Pyramidbox.pytorch
face-detection pyramidbox vgg
Last synced: 2 months ago
JSON representation
Pyramidbox implement with pytorch
- Host: GitHub
- URL: https://github.com/yxlijun/Pyramidbox.pytorch
- Owner: yxlijun
- Created: 2018-12-31T08:26:14.000Z (almost 6 years ago)
- Default Branch: master
- Last Pushed: 2019-03-22T11:20:18.000Z (over 5 years ago)
- Last Synced: 2024-07-21T22:36:25.421Z (2 months ago)
- Topics: face-detection, pyramidbox, vgg
- Language: Python
- Homepage:
- Size: 2.47 MB
- Stars: 155
- Watchers: 7
- Forks: 47
- Open Issues: 18
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
## PyramidBox: A Context-assisted Single Shot Face Detector.##
[A PyTorch Implementation of PyramidBox](https://arxiv.org/abs/1803.07737)
### Description
I train pyramidbox with pytorch and the result approaches the original paper result,the pretrained model can be downloaded in [vgg](https://pan.baidu.com/s/1Q-YqoxJyqvln6KTcIck1tQ),the final model can be downloaded in [Pyramidbox](https://pan.baidu.com/s/1VtzgB9srkJY4SUtVM3n8tw).the AP in WIDER FACE as following:
| | Easy MAP | Medium MAP | hard MAP |
| -------- | ---------|------------| --------- |
| origin paper| 0.960 | 0.948 | 0.888 |
| this repo | 0.948 | 0.938 | 0.880 |
the AP in AFW,PASCAL,FDDB as following:
| AFW | PASCAL | FDDB |
| --------- |-----------| ---------|
| 99.65 | 99.02 | 0.983 |
the gap is small with origin paper,I train 120k batch_size 4 which is different from paper,which maybe cause the gap,if you have more gpu ,the final result maybe better.
### Requirement
* pytorch 0.3
* opencv
* numpy
* easydict
### Prepare data
1. download WIDER face dataset
2. modify data/config.py
3. ``` python prepare_wider_data.py```
### Train
```
python train.py --batch_size 4
--lr 5e-4
```
### Evalution
according to yourself dataset path,modify data/config.py
1. Evaluate on AFW.
```
python tools/afw_test.py
```
2. Evaluate on FDDB
```
python tools/fddb_test.py
```
3. Evaluate on PASCAL face
```
python tools/pascal_test.py
```
4. test on WIDER FACE
```
python tools/wider_test.py
```
### Demo
you can test yourself image
```
python demo.py
```
### Result
### References
* [PyramidBox: A Context-assisted Single Shot Face Detector](https://arxiv.org/abs/1803.07737)
* [PyramidBox model](https://github.com/PaddlePaddle/models/tree/develop/fluid/PaddleCV/face_detection)
* [ssd.pytorch](https://github.com/amdegroot/ssd.pytorch)