1 Star 0 Fork 35

CpDec/FastDeploy

forked from PaddlePaddle/FastDeploy 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

YOLOv5Lite准备部署模型

导出ONNX模型

  • 自动获取 访问YOLOv5Lite 官方github库,按照指引下载安装,下载yolov5-lite-xx.onnx 模型(Tips:官方提供的ONNX文件目前是没有decode模块的)

    #下载yolov5-lite模型文件(.onnx)
    Download from https://drive.google.com/file/d/1bJByk9eoS6pv8Z3N4bcLRCV3i7uk24aU/view
    官方Repo也支持百度云下载
    
  • 手动获取

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

    • 导出含有decode模块的ONNX文件

    首先需要参考YOLOv5-Lite#189的解决办法,修改代码。

    #下载yolov5-lite模型文件(.pt)
    Download from https://drive.google.com/file/d/1oftzqOREGqDCerf7DtD5BZp9YWELlkMe/view
    官方Repo也支持百度云下载
    
    # 导出onnx格式文件
    python export.py --grid --dynamic --concat --weights PATH/TO/yolov5-lite-xx.pt
    
    
    
    • 导出无decode模块的ONNX文件(不需要修改代码)
    #下载yolov5-lite模型文件
    Download from https://drive.google.com/file/d/1oftzqOREGqDCerf7DtD5BZp9YWELlkMe/view
    官方Repo也支持百度云下载
    
    # 导出onnx格式文件
    python export.py --grid --dynamic --weights PATH/TO/yolov5-lite-xx.pt
    
    

下载预训练ONNX模型

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

模型 大小 精度
YOLOv5Lite-e 3.1MB 35.1%
YOLOv5Lite-s 6.3MB 42.0%
YOLOv5Lite-c 18MB 50.9%
YOLOv5Lite-g 21MB 57.6%

详细部署文档

版本说明

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

搜索帮助