README.md

English | 简体中文

YOLOv5Face Ready-to-deploy Model

• 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.

Export ONNX Model

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

Download Pre-trained ONNX Model

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

Detailed Deployment Tutorials

• Python Deployment
• C++ Deployment

Release Note

• This tutorial and related code are written based on YOLOv5Face CommitID:4fd1ead