English | 简体中文

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%