English | 简体中文
- YOLOv5Face
- (1)The *.pt provided by the Official Library can be deployed after the Export ONNX Model to complete the deployment;
- (2)As for YOLOv5Face model trained on customized data, please follow the operations guidelines in Export ONNX Model to complete the deployment.
Visit YOLOv5Face official github repository, follow the guidelines to download the yolov5s-face.pt
model, and employ export.py
to get the file in onnx
format.
-
Download yolov5face model files
Link: https://pan.baidu.com/s/1fyzLxZYx7Ja1_PCIWRhxbw Link: eq0q https://drive.google.com/file/d/1zxaHeLDyID9YU4-hqK7KNepXIwbTkRIO/view?usp=sharing
-
Export files in onnx format
PYTHONPATH=. python export.py --weights weights/yolov5s-face.pt --img_size 640 640 --batch_size 1
-
onnx model simplification (optional)
onnxsim yolov5s-face.onnx yolov5s-face.onnx
For developers' testing, models exported by YOLOv5Face 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 |
---|---|---|
YOLOv5s-Face | 30MB | 94.3 |
YOLOv5s-Face-bak | 30MB | - |
YOLOv5l-Face | 181MB | 95.8 |
- This tutorial and related code are written based on YOLOv5Face CommitID:4fd1ead