Downloads · 30 days
6
17% of all-time downloads
retina-backbone_resnet50-ft_widerface
HaohuaLv/retina-backbone_resnet50-ft_widerface
retina-backbone_resnet50-ft_widerface is a machine learning model from HaohuaLv. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
A retinaface model for Face Detection trained on widerface dataset.
All-time downloads
35
Public
Repo size
223 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin111 MB · 98%
At a glance
- Library
- transformers
- Access
- Public
- Created
- Dec 24, 2023
- Updated
- Dec 24, 2023
- SHA
- 99bc4053
Model card
From the Hugging Face model README
Retinaface Face Detection
A retinaface model for Face Detection trained on widerface dataset.
Notice: This is face detection model's training, evaluation and inference scripts in HuggingFace🤗 style from scratch for practice.
Train
Run
python train.py --model_config_file <MODEL_CONFIG_FILE>
<MODEL_CONFIG_FILE> can be found in folder config.
Model checkpoints will be saved in folder checkpoints by default.
backbone-ResNet50 checkpoint can be download in my Google Drive or HuggingFace🤗.
Inference
Observe logits map and predicted bboxes
Run
python inference.py --checkpoint_path <CHECKPOINT_PATH>
<CHECKPOINT_PATH> is a model folder containing config.json and pytorch_model.bin.
Detect
Run
python detect.py --checkpoint_path <CHECKPOINT_PATH> --image_path <IMAGE_PATH> --save_path <SAVE_PATH>
References
- Library
- transformers
- Created
- Dec 24, 2023
- Updated
- Dec 24, 2023