yolov5 6.1官方模板TensorRT加速保姆级图文教程

加速的基础流程今年春季已经写好了，可以完美复现加速。

需要补充的是运行export.py的报错，加一句：

pip install onnx

这会写教程的时候，电脑是昨天新做的系统，没有python以及对应的工具，这里需要安装：

下载8.2以上的tensorrt版本，个人电脑是i7,1060，下载了TensorRT-8.4.0.6并解压到任意文件夹，打开里面的python目录，找到：

下载官方模板代码，安装到了d盘：

此刻电脑还是干净的，仅仅安装了上面的图片里面的工具：annaconda，清华源和pycharm

打开annaconda的 powershell工具，准备依次键入配置代码，这个过程可能需要半个小时左右：

【yolov5 6.1官方模板TensorRT加速保姆级图文教程】
https://blog.csdn.net/lao_5/article/details/127813970?spm=1001.2014.3001.5502#最简单的方法如下：
conda create -n yoloTRT python=3.8    #创建环境
conda activate yoloTRT    #激活环境
cd D:\yolov5-6.1  #进入程序文件夹
pip install --upgrade setuptools pip wheel  #这里报错,不影响。next。。。
python -m pip install nvidia-pyindex
python -m pip install nvidia-cuda-runtime-cu11
conda install cuda -c nvidia/label/cuda-11.7.0 -c nvidia/label/cuda-11.7
conda install pytorch torchvision torchaudio cudatoolkit=11 -c pytorch
pip install tensorrt-8.4.0.6-cp38-none-win_amd64.whl
pip install onnx
pip install -r requirements.txt
conda deactivate  #退出虚拟环境

顺序复制，静待安装即可，安装过程不做赘述，没有波澜，无脑CV即可完成。

注意多加一个pip install onnx，当然也可以注释代码里的onnx的生成。

这里默认小帅们已经pip了所有的过程，进入下一步，给pycharm定位解释器，也就是刚才部署的

yoloTRT环境中exe即可。这些装备工作就绪之后，6.1模板有一个致命的错误，先修改过来，看图

照做即可：

准备官方的权重文件，我这里下崽了yolov5s.pt做演示用，丢进代码目录，右键执行detect.py,

这样做的意图是确保解释器与代码都是没有问题的。执行看看吧：

很显然，上面的配置代码至关重要，很顺利，一次过了。

接着步入加速，首先要得到一个engine文件，6.1模板用export.py可以直接转换，需要改的地方，是输出engine文件名以及在export.py中搜索device，然后把cpu改为0

这样模板代码目录就多出一个yolov5s.engine文件，当然还有onnx和torchscript，这两位暂时不用

继续转到detect.py页面，把代码里面的yolov5s.pt改为yolov5s.engine，到这里大功告成了，右键

运行代码，结果如图：

关于加速，就彻底成功了。detect.py里面的代码但是精华，仔细研读理解可以做你爱做的事了。

yolo交流群：

967082372

yolov5 6.1官方模板TensorRT加速保姆级图文教程相关推荐

Promox VE安装黑群晖保姆级图文教程
本教程是PVE家用All in one一体机方案的保姆级攻略之一,将详细介绍在PVE虚拟环境用xpenology大神Jun的引导镜像引导安装黑群晖的详细过步骤. PVE虚拟环境的构建请移步:Proxm ...
Alexa对接开发java版(保姆级图文教程)
文章目录前言一.准备工作? 二.使用步骤 1.登陆科技账号创建函数 2.登陆开发在账号创建技能 3.测试总结前言提示:真正的保姆级图文教程,因本人java开发,准备写篇java+node版本 ...
Mysql读写锁保姆级图文教程
摘要:读锁会阻塞写,但是不会阻塞读,而写锁会把杜希俄都阻塞. 本文分享自华为云社区<Mysql保姆级读写锁图文教程丨[绽放吧!数据库]>,作者:Code皮皮虾 . 准备创建mylock表 ...
[保姆级图文教程]如何配置局域网多机共享同步文件夹
[图文教程]如何配置局域网多机共享同步文件夹 1.控制面板设置 2.开启共享设置 3.本地组策略编辑器 4.大功告成 1.控制面板设置控制面板>>网络和Internet>>网 ...
应粉丝需求力肝App爬虫入门教学，保姆级图文教程，⭐建议收藏⭐！！!
文章目录前言环境准备夜神模拟器 Fiddler 配置 Fiddler配置安装证书入门爬取爬虫案例最后前言网页爬虫相信小伙伴们都会,以倒霉蛋百度为例,进入首页,直接F12打开控制台 ...
WordPress网站更改后台登录地址保姆级图文教程
背景大家都知道WordPress系统默认的后台地址是域名/wp-admin 或域名/wp-login.php 这就意味着所有用WordPress的人都可以用这个地址打开你的后台登录页面,很明显谁都 ...
ROS从入门到精通0-2：Win10+Ubuntu双系统安装、配置、卸载保姆级图文教程
目录 0 电脑环境 1 安装Ubuntu双系统 1.1 刻录系统引导盘 1.2 在Windows下创建空白分区 1.3 重启电脑进入安装界面 1.4 图形化安装Ubuntu 2 配置Ubuntu系统 ...
alexa skill+自定义oauth2服务完整版教程(保姆级图文教程附demo源码)
文章目录前言一.alexa skill是什么? 二.开发步骤 1.注册账号 2.创建技能 3.创建函数技能绑定函数部署oauth2 为技能配置账号,开启oauth2认证 alexa app 开 ...
前端开发条形码生成、打印的2种方式总结（保姆级图文教程）
文章目录前言一.前端打印的2方式二.使用步骤 1. 打印机调试 2. 浏览器打印生成二维码插件JsBarcode 使用浏览器默认打印代码 3. 第三方托盘程序打印托盘打印插件详细步骤问题 ...

yolov5 6.1官方模板TensorRT加速保姆级图文教程

yolov5 6.1官方模板TensorRT加速保姆级图文教程相关推荐

最新文章

热门文章