Skip to content

Latest commit

 

History

History
 
 

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

YOLOv5Cls准备部署模型

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

下载预训练ONNX模型

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

模型 大小 精度(top1) 精度(top5)
YOLOv5n-cls 9.6MB 64.6% 85.4%
YOLOv5s-cls 21MB 71.5% 90.2%
YOLOv5m-cls 50MB 75.9% 92.9%
YOLOv5l-cls 102MB 78.0% 94.0%
YOLOv5x-cls 184MB 79.0% 94.4%

详细部署文档

版本说明