Files
FastDeploy/examples/vision/facedet/yolov5face
huangjianhui 376fdbfe2c [Other] Update old Api to new ones (#861)
* Update keypointdetection result docs

* Update im.copy() to im in examples

* Update new Api, fastdeploy::vision::Visualize to fastdeploy::vision

* Update SwapBackgroundSegmentation && SwapBackgroundMatting to SwapBackground

* Update README_CN.md

* Update README_CN.md
2022-12-14 17:25:58 +08:00
..
2022-08-15 20:25:03 +08:00

YOLOv5Face准备部署模型

导出ONNX模型

访问YOLOv5Face官方github库按照指引下载安装下载yolov5s-face.pt 模型,利用 export.py 得到onnx格式文件。

  • 下载yolov5face模型文件

    Link: https://pan.baidu.com/s/1fyzLxZYx7Ja1_PCIWRhxbw Link: eq0q  
    https://drive.google.com/file/d/1zxaHeLDyID9YU4-hqK7KNepXIwbTkRIO/view?usp=sharing
    
  • 导出onnx格式文件

    PYTHONPATH=. python export.py --weights weights/yolov5s-face.pt --img_size 640 640 --batch_size 1  
    
  • onnx模型简化(可选)

    onnxsim yolov5s-face.onnx yolov5s-face.onnx
    

下载预训练ONNX模型

为了方便开发者的测试下面提供了YOLOv5Face导出的各系列模型开发者可直接下载使用。下表中模型的精度来源于源官方库

模型 大小 精度
YOLOv5s-Face 30MB 94.3
YOLOv5s-Face-bak 30MB -
YOLOv5l-Face 181MB 95.8

详细部署文档

版本说明