1 Star 0 Fork 35

CpDec/FastDeploy

forked from PaddlePaddle/FastDeploy 
Create your Gitee Account
Explore and code with more than 13.5 million developers,Free private repositories !:)
Sign up
文件
Clone or Download
contribute
Sync branch
Cancel
Notice: Creating folder will generate an empty file .keep, because not support in Git
Loading...
README

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%

详细部署文档

版本说明

马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/martinfields/FastDeploy.git
git@gitee.com:martinfields/FastDeploy.git
martinfields
FastDeploy
FastDeploy
develop

Search