Skip to content

Latest commit

 

History

History
45 lines (32 loc) · 2.15 KB

File metadata and controls

45 lines (32 loc) · 2.15 KB

English | 简体中文

YOLOv5Face Ready-to-deploy Model

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

Release Note