ljchang commited on
Commit
4178fe6
·
verified ·
1 Parent(s): a0eecb3

Update README.md

Browse files

added initial model card

Files changed (1) hide show
  1. README.md +33 -9
README.md CHANGED
@@ -1,9 +1,33 @@
1
- ---
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]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Retinaface
2
+
3
+ ## Model Details
4
+
5
+ ### Model Description
6
+ 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.
7
+
8
+ - **License:** MIT
9
+ - **License Link:** [MIT License](https://github.com/biubug6/Pytorch_Retinaface/blob/master/LICENSE.MIT)
10
+
11
+ ### Model Sources
12
+ - **Repository:** [Pytorch_Retinaface](https://github.com/biubug6/Pytorch_Retinaface)
13
+ - **Paper:** [RetinaFace: Single-stage Dense Face Localisation in the Wild](https://arxiv.org/abs/1905.00641)
14
+
15
+ ## Model Architecture
16
+ 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.
17
+
18
+ ## Intended Use
19
+ This model is intended for use in applications requiring face detection, such as:
20
+ - Security systems
21
+ - Augmented reality
22
+ - Image processing pipelines
23
+ - Photo management applications
24
+
25
+ ## Citation
26
+ **BibTeX:** @misc{deng2019retinafacesinglestagedenseface,
27
+ title={RetinaFace: Single-stage Dense Face Localisation in the Wild},
28
+ author={Jiankang Deng and Jia Guo and Yuxiang Zhou and Jinke Yu and Irene Kotsia and Stefanos Zafeiriou},
29
+ year={2019},
30
+ eprint={1905.00641},
31
+ archivePrefix={arXiv},
32
+ primaryClass={cs.CV},
33
+ url={https://arxiv.org/abs/1905.00641