Update README.md
Browse files
README.md
CHANGED
@@ -2,8 +2,60 @@
|
|
2 |
tags:
|
3 |
- model_hub_mixin
|
4 |
- pytorch_model_hub_mixin
|
|
|
|
|
|
|
5 |
---
|
6 |
|
7 |
-
|
8 |
-
|
9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
tags:
|
3 |
- model_hub_mixin
|
4 |
- pytorch_model_hub_mixin
|
5 |
+
license: mit
|
6 |
+
language:
|
7 |
+
- en
|
8 |
---
|
9 |
|
10 |
+
# Retinaface
|
11 |
+
|
12 |
+
## Model Details
|
13 |
+
|
14 |
+
### Model Description
|
15 |
+
Retinaface is a state-of-the-art face detection model built using PyTorch. It accurately detects faces in images and returns bounding boxes around detected faces. The model is designed to work efficiently on a wide range of images, including those with varying lighting conditions, occlusions, and face orientations.
|
16 |
+
|
17 |
+
- **License:** MIT
|
18 |
+
- **License Link:** [MIT License](https://github.com/biubug6/Pytorch_Retinaface/blob/master/LICENSE.MIT)
|
19 |
+
|
20 |
+
### Model Sources
|
21 |
+
- **Repository:** [Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface)
|
22 |
+
- **Paper:** [RetinaFace: Single-stage Dense Face Localisation in the Wild](https://arxiv.org/abs/1905.00641)
|
23 |
+
|
24 |
+
## Model Architecture
|
25 |
+
The Retinaface model utilizes a deep convolutional neural network architecture with multiple layers. It uses `mobilenet0.25` as the backbone network (only 1.7M parameters) but can also use `resnet50` as the backbone to achieve better results. It includes additional layers for feature extraction and bounding box prediction.
|
26 |
+
|
27 |
+
## Intended Use
|
28 |
+
This model is intended for use in applications requiring face detection, such as:
|
29 |
+
- Security systems
|
30 |
+
- Augmented reality
|
31 |
+
- Image processing pipelines
|
32 |
+
- Photo management applications
|
33 |
+
|
34 |
+
## Evaluation Results
|
35 |
+
The model was evaluated on the WIDER FACE dataset
|
36 |
+
|
37 |
+
## Limitations and Biases
|
38 |
+
While the Retinaface model performs well in many conditions, it may have limitations, including:
|
39 |
+
|
40 |
+
Reduced accuracy in detecting faces with heavy occlusions
|
41 |
+
Potential biases towards the demographic distribution of the training dataset
|
42 |
+
Ethical Considerations
|
43 |
+
When using face detection technologies, it is essential to consider the ethical implications, such as privacy concerns and potential biases. Ensure that the use of this model complies with relevant regulations and guidelines.
|
44 |
+
|
45 |
+
## Citation
|
46 |
+
If you use the Retinaface model in your research or application, please cite the following paper:
|
47 |
+
|
48 |
+
```
|
49 |
+
@misc{deng2019retinafacesinglestagedenseface,
|
50 |
+
title={RetinaFace: Single-stage Dense Face Localisation in the Wild},
|
51 |
+
author={Jiankang Deng and Jia Guo and Yuxiang Zhou and Jinke Yu and Irene Kotsia and Stefanos Zafeiriou},
|
52 |
+
year={2019},
|
53 |
+
eprint={1905.00641},
|
54 |
+
archivePrefix={arXiv},
|
55 |
+
primaryClass={cs.CV},
|
56 |
+
url={https://arxiv.org/abs/1905.00641}
|
57 |
+
}
|
58 |
+
```
|
59 |
+
|
60 |
+
## Acknowledgements
|
61 |
+
We thank the contributors and the open-source community for their valuable support in developing this model. Special thanks to the authors of the original Retinaface paper and the WIDER FACE dataset.
|