ljchang commited on
Commit
b275f0e
·
verified ·
1 Parent(s): 9ceddc9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +55 -3
README.md CHANGED
@@ -2,8 +2,60 @@
2
  tags:
3
  - model_hub_mixin
4
  - pytorch_model_hub_mixin
 
 
 
5
  ---
6
 
7
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
- - Library: [More Information Needed]
9
- - Docs: [More Information Needed]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.