mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 16:48:03 +08:00
cbf88a46fa
* 第一次提交 * 补充一处漏翻译 * deleted: docs/en/quantize.md * Update one translation * Update en version * Update one translation in code * Standardize one writing * Standardize one writing * Update some en version * Fix a grammer problem * Update en version for api/vision result * Merge branch 'develop' of https://github.com/charl-u/FastDeploy into develop * Checkout the link in README in vision_results/ to the en documents * Modify a title * Add link to serving/docs/ * Finish translation of demo.md * Update english version of serving/docs/ * Update title of readme * Update some links * Modify a title * Update some links * Update en version of java android README * Modify some titles * Modify some titles * Modify some titles * modify article to document * update some english version of documents in examples * Add english version of documents in examples/visions * Sync to current branch * Add english version of documents in examples * Add english version of documents in examples * Add english version of documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples * Update some documents in examples
English | 简体中文
RetinaFace Ready-to-deploy Model
- RetinaFace
- （1）The *.pt provided by theOfficial Library can be deployed after the Export ONNX Model；
- （2）As for RetinaFace model trained on customized data, please follow the Export ONNX Model to complete the deployment.
Export ONNX Model
Download the pre-trained ONNX modelThe model has been converted to ONNX. If you downloaded the model from the RetinaFace official repo, please follow the tutorial below to export ONNX.
- Download the official repository
git clone https://github.com/biubug6/Pytorch_Retinaface.git
- Download the pre-trained weights and place them in the weights folder
./weights/
mobilenet0.25_Final.pth
mobilenetV1X0.25_pretrain.tar
Resnet50_Final.pth
- run convert_to_onnx.py to export ONNX model files
PYTHONPATH=. python convert_to_onnx.py --trained_model ./weights/mobilenet0.25_Final.pth --network mobile0.25 --long_side 640 --cpu
PYTHONPATH=. python convert_to_onnx.py --trained_model ./weights/Resnet50_Final.pth --network resnet50 --long_side 640 --cpu
Attention: We need to add a type constraint, type=int, to the --long_side parameter in the convert_to_onnx.py script.
- Use onnxsim to simplify the model
onnxsim FaceDetector.onnx Pytorch_RetinaFace_mobile0.25-640-640.onnx # mobilenet
onnxsim FaceDetector.onnx Pytorch_RetinaFace_resnet50-640-640.onnx # resnet50
Download pre-trained ONNX models
For developers' testing, models exported by RetinaFace are provided below. Developers can download and use them directly. (The accuracy of the models in the table is sourced from the official library)
| Model | Size | Accuracy |
|---|---|---|
| RetinaFace_mobile0.25-640 | 1.7MB | - |
| RetinaFace_mobile0.25-720 | 1.7MB | - |
| RetinaFace_resnet50-640 | 105MB | - |
| RetinaFace_resnet50-720 | 105MB | - |
Detailed Deployment Tutorials
Release Note
- This tutorial and related code are written based on RetinaFace CommitID:b984b4b