Files
FastDeploy/examples/vision/detection/yolov5
WJJ1995 b557dbc2d8 Add YOLOv5-cls Model (#335)
* add yolov5cls

* fixed bugs

* fixed bugs

* fixed preprocess bug

* add yolov5cls readme

* deal with comments

* Add YOLOv5Cls Note

* add yolov5cls test

Co-authored-by: Jason <jiangjiajun@baidu.com>
2022-10-12 15:57:26 +08:00
..
2022-09-28 17:45:02 +08:00
2022-10-12 15:57:26 +08:00

YOLOv5准备部署模型

  • YOLOv5 v6.0部署模型实现来自YOLOv5,和基于COCO的预训练模型
    • 1官方库提供的*.onnx可直接进行部署
    • 2开发者基于自己数据训练的YOLOv5 v6.0模型,可使用YOLOv5中的export.py导出ONNX文件后完成部署。

下载预训练ONNX模型

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

模型 大小 精度
YOLOv5n 1.9MB 28.4%
YOLOv5s 7.2MB 37.2%
YOLOv5m 21.2MB 45.2%
YOLOv5l 46.5MB 48.8%
YOLOv5x 86.7MB 50.7%

详细部署文档

版本说明