https://github.com/jahongir7174/facedet-onnx
Sample and Computation Redistribution for Efficient Face Detection
https://github.com/jahongir7174/facedet-onnx
face-detection facedetection inference onnx onnxruntime opencv opencv-python python python3 scrfd test
Last synced: 9 days ago
JSON representation
Sample and Computation Redistribution for Efficient Face Detection
- Host: GitHub
- URL: https://github.com/jahongir7174/facedet-onnx
- Owner: jahongir7174
- Created: 2021-01-31T11:05:49.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2023-02-14T02:44:28.000Z (over 2 years ago)
- Last Synced: 2025-04-10T14:16:38.150Z (6 months ago)
- Topics: face-detection, facedetection, inference, onnx, onnxruntime, opencv, opencv-python, python, python3, scrfd, test
- Language: Python
- Homepage: https://arxiv.org/abs/2105.04714
- Size: 73.7 MB
- Stars: 5
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
Inference code of [SCRFD](https://arxiv.org/abs/2105.04714) using ONNX Runtime
### Installation
```
conda create -n ONNX python=3.8
conda activate ONNX
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
pip install onnxruntime-gpu==1.14.0
pip install opencv-python==4.5.5.64
```
### Test
* Run `python main.py $1 $2` for testing, `$1` is model file path and `$2` is image file path
### Note
* This repo supports inference only, see reference for more details
* See `./weights/*` folder to check list of available models, `_kps` means the model includes 5 keypoints prediction
#### Reference
* https://github.com/deepinsight/insightface/tree/master/detection/scrfd