https://github.com/jahongir7174/scrfd-tiny
https://github.com/jahongir7174/scrfd-tiny
Last synced: 6 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/jahongir7174/scrfd-tiny
- Owner: jahongir7174
- Created: 2025-02-02T11:10:27.000Z (8 months ago)
- Default Branch: master
- Last Pushed: 2025-02-03T05:31:16.000Z (8 months ago)
- Last Synced: 2025-02-03T06:27:28.372Z (8 months ago)
- Language: Python
- Size: 968 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
Tiny version of [SCRFD](https://arxiv.org/abs/2105.04714) Face and 5 Point Landmark Detection implementation using PyTorch
### Installation
```
conda create -n PyTorch python=3.11.11
conda activate PyTorch
conda install python=3.11.11 pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
pip install opencv-python
pip install PyYAML
pip install tqdm
```
### Train
* Configure your dataset path in `main.py` for training
* Run `bash main.sh $ --train` for training, `$` is number of GPUs
### Test
* Configure your dataset path in `main.py` for testing
* Run `python main.py --test` for testing
### Results
| Model | AP-easy | AP-medium | AP-hard | Params (M) | Params Ratio | FLOPS (M) | CPU ONNX Latency (ms) |
|--------------------|---------|-----------|---------|------------|--------------|-----------|-----------------------|
| SCRFD0.5(ICLR2022) | 90.57 | 88.12 | 68.51 | 0.57 | 1.00x | 508 | 5.5 |
| SCRFD-tiny(ours) | 89.23 | 87.60 | 67.15 | 0.08 | 7.12x | 451 | 4.4 |
### Dataset structure
├── WIDERFace
├── images
├── train
├── 1111.jpg
├── 2222.jpg
├── val
├── 1111.jpg
├── 2222.jpg
├── labels
├── train
├── 1111.txt
├── 2222.txt
├── val
├── 1111.txt
├── 2222.txt
#### Reference
* https://github.com/jahongir7174/YUNet-eval
* https://github.com/ShiqiYu/libfacedetection.train
* https://github.com/deepinsight/insightface/tree/master/detection/scrfd