yoloV8导出engine模型

news2024/9/20 22:36:54

yoloV8导出engine模型

引言:
目的是TensorRT加速YOLO实例分割模型。
本博客记录达成此目的所需的步骤,及步骤中可能遇到的问题。
as follow:

1. 首先导出onnx模型的脚本命令:

yolo export model=best.pt format=onnx opset=12 simplify=True

脚本命令的参数:
yolo导出模型参数
不同格式使用的模型对应:
在这里插入图片描述
我遇到的几个问题:

ImportError: DLL load failed while importing _imaging: 找不到指定的模块。
解决:pip卸载Pillow,再重新安装

pip uninstall Pillow
pip install Pillow

ModuleNotFoundError: No module named 'onnx'
解决:pip安装onnx

pip install onnx

requirements: Ultralytics requirements ['cmake', 'onnxsim>=0.4.33', 'onnxruntime'] not found, attempting AutoUpdate...
requirements: ❌ AutoUpdate skipped (offline)
解决:pip安装cmake、onnxsim【版本>=0.4.33,可能遇到文件名过长的问题】、onnxruntime
在这里插入图片描述

pip install cmake

pip install onnxsim==0.4.33 #可能会报错文件名过长【原因可能是安装python的时候没有禁用文件名长度限制】
#进入conda虚拟环境后
mountvol -x D:\ /s
#然后再次
pip install onnxsim==0.4.33

pip install onnxruntime

最后导出best.onnx模型:
在这里插入图片描述

2. 先测试一下导出onnx模型【脚本命令】:

yolo task=segment mode=predict model=best.onnx source=18.JPG show=True save=True device=0

在这里插入图片描述
测试效果:
在这里插入图片描述

3. TensorRT安装、onnx转engine:

TensorRT官网:https://developer.nvidia.com/tensorrt/download
在这里插入图片描述
在这里插入图片描述
安装教程【参考博客】:https://blog.csdn.net/qq_44747572/article/details/129022225
版本对应【参考博客】:
https://blog.csdn.net/initative/article/details/138213992

安装完成后,可以使用trtexec 命令:

trtexec --onnx=best.onnx --saveEngine=best.engine

即可得到yolo的engine模型。
trtexec命令参数【参考博客】:
https://blog.csdn.net/HW140701/article/details/120360642

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1923477.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

全网都在找数学建模美赛模版,免费直接复制,无需关注公众号!!!

文件夹结构如下 美版论文模版预览(截选) tex文件代码如下 \documentclass{mcmthesis} \mcmsetup{CTeX false, % 使用 CTeX 套装时,设置为 truetcn 1111111, problem ABCDEF,sheet true, titleinsheet true, keywordsinsheet true,titlepage false, abstra…

C++ | Leetcode C++题解之第226题翻转二叉树

题目: 题解: class Solution { public:TreeNode* invertTree(TreeNode* root) {if (root nullptr) {return nullptr;}TreeNode* left invertTree(root->left);TreeNode* right invertTree(root->right);root->left right;root->right …

深度学习5 神经网络

生物神经网络是指人的大脑,这是人工神经网络的技术原型。根据生物神经网络的原理,人们用计算机复现了简化的神经网络。当然,人工神经网络是机器学习的一大分支。 1.基本组成 1.1神 经 元 神经元是神经网络的基本组成。激活函数又称作激励函…

【java报错已解决】“Array Out of Bounds“

🎬 鸽芷咕:个人主页 🔥 个人专栏: 《C干货基地》《粉丝福利》 ⛺️生活的理想,就是为了理想的生活! 文章目录 引言一、问题描述1.1 报错示例1.2 报错分析1.3 解决思路 二、解决方法:2.1 方法一:范围检查2.…

案例效果展示小程序制作承载信息宣传

如工商财税、房地产、装修、鲜花、五金等很多行业,都有商品/服务/案例/效果等展示宣传获客的需求,传统线下门店/传单方式效果比较低,线上渠道则更多更利于商家随时开展。 然而更多的图片和文字介绍分享混乱难以整理且无法全面覆盖&#xff0…

如何利用人工智能大模型分析流量来源,提升数字化营销效果

摘要 流量来源分析是数字化营销中的一个重要环节,它可以帮助产品经理和运营人员了解用户从哪里来,为什么来,以及如何留住他们。流量来源分析可以分为三大类:搜索引擎流量,社交媒体流量,和广告投放流量。每…

代理模式(大话设计模式)C/C++版本

代理模式 C #include <iostream> using namespace std;class Subject // Subject 定义了RealSubject和Proxy的共用接口..这样就在任何使用RealSubject的地方都可以使用Proxy { public:virtual void func(){cout << "Subject" << endl;} };class R…

SQL 将titles_test表名修改为titles_2017

系列文章目录 文章目录 系列文章目录前言 前言 前些天发现了一个巨牛的人工智能学习网站&#xff0c;通俗易懂&#xff0c;风趣幽默&#xff0c;忍不住分享一下给大家。点击跳转到网站&#xff0c;这篇文章男女通用&#xff0c;看懂了就去分享给你的码吧。 描述 将titles_tes…

2024年辽宁省数学建模竞赛B题超详细解题思路+问题一代码分享

B题相关优秀论文问题一数据预处理代码分享链接&#xff1a; https://pan.baidu.com/s/1wpNZYW5v8jf2iM0uB7pDEA 提取码&#xff1a;sxjm B题 钢铁产品质量优化 研究现状 冷轧带钢是钢铁企业的高附加值产品&#xff0c;其产品质量稳定性对于钢铁企业的经 济效益具有非常重要…

SpringBoot整合JAX-RS接口

目录 二、创建RESTful资源 三、注册JAX-RS资源 四、修改配置等信息 五、启动SpringBoot程序、访问服务 六、遇到的问题 七、与feign进行配合使用 1、接口定义 2、接口实现 3、注册资源 4、调用方web服务实现&#xff0c;跟注入普通服务一样 5、启动两个服务&#xff…

很多人误解:谷歌账号就是谷歌邮箱账号吗?Google账号=Gmail?

在服务的过程中&#xff0c;我发现有一些朋友对于什么是Google账号、谷歌账号和gmail地址&#xff0c;谷歌邮箱和辅助邮箱等等的概念和用法搞得不够清爽&#xff0c;使得偶会会出现问题&#xff0c;甚至给谷歌账号带来一些风险。所以这篇文章&#xff0c;GG账号服务为大家解析一…

LeetCode刷题之HOT100之打家劫舍Ⅲ

2024 7/13 七月就要过去一半了&#xff0c;伴随着蝉鸣&#xff0c;酷暑被窗户隔离&#xff0c;我坐在凳子上&#xff0c;有一只蚊子不愿意放过我。 1、题目描述 2、算法分析 前面做过打家劫舍Ⅱ&#xff0c;今天来做打家劫舍Ⅲ&#xff0c;小偷也是越来越聪明了&#xff0c;意…

集训 Day 3 总结 虚树 + dfs tree + 基环树

虚树 虚树&#xff0c;顾名思义是 只关注原树上的某些 关键点&#xff0c;在保留原树祖孙关系的前提下建出的一棵边数、点数大大减少的树 适用于优化某些在整棵树上进行 d p dp dp、 d f s dfs dfs 的问题 通常是题目中出现多次询问&#xff0c;每次给出树上的一些关键点&a…

唯一索引插入引发的死锁问题

MySQL 8.4.0 RR隔离级别 场景复现 分析下面SQL死锁的场景 对于switch表&#xff0c;有主键索引id和唯一索引(uid、type)。 CREATE TABLE switch (id int NOT NULL AUTO_INCREMENT COMMENT Unique identifier for each switch,uid int NOT NULL COMMENT User ID associated wi…

Autoware 定位之基于ARTag的landmark定位(六)

Tip: 如果你在进行深度学习、自动驾驶、模型推理、微调或AI绘画出图等任务&#xff0c;并且需要GPU资源&#xff0c;可以考虑使用UCloud云计算旗下的Compshare的GPU算力云平台。他们提供高性价比的4090 GPU&#xff0c;按时收费每卡2.6元&#xff0c;月卡只需要1.7元每小时&…

AI大模型来了,低代码还有机会吗?

AI大模型的突飞猛进&#xff0c;不仅引领了技术的革新浪潮&#xff0c;也为各行各业的发展带来了前所未有的挑战与机遇。近年来&#xff0c;随着人工智能技术的不断进步&#xff0c;关于各行各业将被AI取代的论调此起彼伏&#xff0c;引发了许多从业者的不安。 几年前&#xf…

Flowable-流程图标与流程演示

BPMN 2.0是业务流程建模符号2.0的缩写。它由Business Process Management Initiative这个非营利协会创建并不断发展。作为一种标识&#xff0c;BPMN 2.0是使用一些符号来明确业务流程设计流程图的一整套符号规范&#xff0c;它能增进业务建模时的沟通效率。目前BPMN2.0是最新的…

AI人工智能填词,唱响心中独特旋律

在音乐的无垠宇宙中&#xff0c;每个人的内心都有一段独一无二的旋律在悄然回荡。而如今&#xff0c;人工智能填词正以其神奇的力量&#xff0c;帮助我们将这些深藏心底的旋律化作动人的歌词&#xff0c;让它们得以放声歌唱。 “妙笔生词智能写歌词软件&#xff08;veve522&am…

【QT】布局管理器

布局管理器 布局管理器1. 垂直布局2. 水平布局3. 网格布局4. 表单布局5. Spacer 布局管理器 之前使⽤ Qt 在界⾯上创建的控件, 都是通过 “绝对定位” 的⽅式来设定的&#xff1b;也就是每个控件所在的位置, 都需要计算坐标, 最终通过 setGeometry 或者 move ⽅式摆放过去。 …

一文彻底学会Vue3路由:全面讲解路由流程、路由模式、传参等——全栈开发之路--前端篇(7)路由详解

全栈开发一条龙——前端篇 第一篇&#xff1a;框架确定、ide设置与项目创建 第二篇&#xff1a;介绍项目文件意义、组件结构与导入以及setup的引入。 第三篇&#xff1a;setup语法&#xff0c;设置响应式数据。 第四篇&#xff1a;数据绑定、计算属性和watch监视 第五篇 : 组件…