ljchang commited on
Commit
2086fc4
1 Parent(s): 6d64f2e

Upload Retinaface Model

Browse files
Files changed (2) hide show
  1. README.md +3 -53
  2. model.safetensors +1 -1
README.md CHANGED
@@ -2,58 +2,8 @@
2
  tags:
3
  - model_hub_mixin
4
  - pytorch_model_hub_mixin
5
- license: mit
6
  ---
7
 
8
- # Retinaface
9
-
10
- ## Model Details
11
-
12
- ### Model Description
13
- 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.
14
-
15
- - **License:** MIT
16
- - **License Link:** [MIT License](https://github.com/biubug6/Pytorch_Retinaface/blob/master/LICENSE.MIT)
17
-
18
- ### Model Sources
19
- - **Repository:** [Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface)
20
- - **Paper:** [RetinaFace: Single-stage Dense Face Localisation in the Wild](https://arxiv.org/abs/1905.00641)
21
-
22
- ## Model Architecture
23
- 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.
24
-
25
- ## Intended Use
26
- This model is intended for use in applications requiring face detection, such as:
27
- - Security systems
28
- - Augmented reality
29
- - Image processing pipelines
30
- - Photo management applications
31
-
32
- ## Evaluation Results
33
- The model was evaluated on the WIDER FACE dataset
34
-
35
- ## Limitations and Biases
36
- While the Retinaface model performs well in many conditions, it may have limitations, including:
37
-
38
- Reduced accuracy in detecting faces with heavy occlusions
39
- Potential biases towards the demographic distribution of the training dataset
40
- Ethical Considerations
41
- 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.
42
-
43
- ## Citation
44
- If you use the Retinaface model in your research or application, please cite the following paper:
45
-
46
- ```
47
- @misc{deng2019retinafacesinglestagedenseface,
48
- title={RetinaFace: Single-stage Dense Face Localisation in the Wild},
49
- author={Jiankang Deng and Jia Guo and Yuxiang Zhou and Jinke Yu and Irene Kotsia and Stefanos Zafeiriou},
50
- year={2019},
51
- eprint={1905.00641},
52
- archivePrefix={arXiv},
53
- primaryClass={cs.CV},
54
- url={https://arxiv.org/abs/1905.00641}
55
- }
56
- ```
57
-
58
- ## Acknowledgements
59
- 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.
 
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]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8a71d965f0f2a8e83b0a20d4ba18848f4373c786b7bdcf9e11e9d68b0a38f867
3
  size 1760200
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:142809e736fa8da810df36de899de14f78753f44d323dd8a910c5cedfa2d284f
3
  size 1760200