Ubuntu 18.04 TensorRT 加速YOLOv5

已于 2022-05-07 13:33:07 修改
阅读量1.1k 收藏 23
点赞数 3
分类专栏： 目标检测 文章标签： opencv cmake
于 2021-05-21 17:01:23 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_50008473/article/details/117118886
版权

环境准备：https://blog.csdn.net/weixin_50008473/article/details/115250986?spm=1001.2014.3001.5501

1. 使用TensorRT推理图片

（1）下载yolov5和 yolov5 tensorrt

git clone https://github.com/wang-xinyu/tensorrtx.git
git clone https://github.com/ultralytics/yolov5.git

（2）将tensorrt/yolov5/gen_wts.py拷贝至yolov5下：
sudo cp -r tenorrtx/yolov5/gen_wts.py yolov5
（3）生成yolov5s.pt…yolov5x.pt对应的yolov5s.wts…yolov5x.wts:

cd yolov5
python gen_wts.py

（4）修改CMakeLists.txt：

# cuda
include_directories(/usr/local/cuda-11.
了解本专栏 订阅专栏 解锁全文 超级会员免费看
June vinvin
关注
  • 3
    点赞
  • 踩
  • 23
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
专栏目录
订阅专栏
[yolov5目标检测] Ubuntu18.04安装tensorrt加速网络推理
weixin_42448226的博客
11-13 1518
1.安装cuda10.0和cudnn7.6.5 dpkg -l | grep cuda 查看cuda安装情况，若发现有libcudnn版本不适应（即安装的cuda版本不是10.0） 使用以下命令删除： sudo dpkg --purge libcudnn7-dev sudo dpkg --purge libcudnn7 2.下载tensorrt的deb文件并安装 参考：nVidia官方安装指导 sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_
【yolov5-代码细读】
最新发布
祖国花朵的博文
05-22 657
知其然更要知其所以然
Ubuntu18.04离线安装g++和make
06-25
g++-9_9.4.0-1ubuntu1~18.04_amd64.deb gcc-9_9.4.0-1ubuntu1~18.04_amd64.deb make_4.1-9.1ubuntu1_amd64.deb
Ubuntu+Yolov5+TensortRT加速部署
张伟的专栏
12-12 761
本文章向大家介绍Ubuntu+Yolov5+TensortRT加速部署，主要包括Ubuntu+Yolov5+TensortRT加速部署使用实例、应用技巧、基本知识点总结和需要注意事项，具有一定的参考价值，需要的朋友可以参考一下。
【Jetson平台 ubuntu 18.04 使用 TensorRT 加速 YOLOv5 检测】
hxb971002的博客
07-19 1460
Jetson平台 ubuntu 18.04 使用 TensorRT 加速 YOLOv5 检测
ubuntu18下TensorRT加速YOLOV5推理（超详细）
qq_51331745的博客
12-31 3446
安装TensorRT TensorRT是Nvidia官方给的C++推理加速工具，如同OpenVINO之于Intel。支持诸多的AI框架，如Tensorflow，Pytorch，Caffe，MXNet等。 可以通过以下链接进行TensorRT的下载安装，根据自己的cuda版本安装相对应的版本。 下载连接：https://developer.nvidia.com/tensorrt 下载这个需要注册nvidia账号，按要求注册就好了。 下载完成后解压： tar -xvzf TensorRT-6.0.1.5.Ubu
ubuntu安装并进行tensorrt加速包YOLOv5推理
m0_46339652的博客
09-06 758
目录1. 下载tensorrt2. 安装tensorrt2.1 安装python-tensorrt2.2 安装uff2.3 查看tensorrt版本3. 生成engine文件3.1 生成wts格式权重文件3.2 cmake编译 1. 下载tensorrt 笔者系统为ubuntu20.04, 由于Nvidia官网上面Tensorrt7.X版本没有找到对应ubuntu20.04的，因此使用ubuntu18.04版本的包进行下载试一试看能不能用: 下载后提取出来。ubuntu下载时选择amd64后缀的。 2.
ubuntu18.04gcc.zip
10-29
ubuntu18.04 离线安装gcc
ubuntu1804 qt调用yolov4源码
02-25
在本项目中，我们主要探讨的是如何在Ubuntu 18.04操作系统上，使用Qt C++框架调用YOLOv4的源码进行对象检测。YOLO（You Only Look Once）是一种流行的实时目标检测算法，而YOLOv4作为其最新版本，在速度和精度上都有...
ubuntu18.04安装net-tools
05-09
ubuntu18.04安装net-tools
在Ubuntu 18.04（AGX板）部署ros2_foxy
07-25
在Ubuntu 18.04系统上部署ROS2 Foxy是一个重要的任务，特别是在AGX板上，这通常用于自动驾驶和机器人应用。以下是一步步指导如何在JetPack 4.6.2环境下完成此过程。 首先，为了提高软件包下载速度，我们需要替换...
yolov5使用TensorRT进行c++部署
weixin_41311686的博客
12-29 6726
有了前面用c++进行opencv里dnn部署和onnxruntime部署的经验，使用TensorRT进行部署，我们只要了解tensorrt和cuda的一些相关api的使用即可方便的部署，整个部署流程都差不多。
ubuntu18.04 yolov5 tensorrt部署
weixin_45866058的博客
08-29 2530
ubuntu18.04 yolov5 tensorrt部署
ubuntu18下使用TensorRT对yolov5加速记录
小俊俊的博客
03-22 4603
ubuntu下使用TensorRT对yolov5加速记录 一，安装TensorRT 1.对应自己的Cuda 与Cudnn版本 cat /usr/local/cuda/version.txt cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 我的版本信息如下 2.安装TensorRT
ubuntu下yolov5 tensorrt模型部署
整理ing
10-11 1595
ubuntu下 yolov5 tensorrt模型部署加速
ubuntu18.04 CUDA10.0 Tensorrtx yolov5部署
qq_41263444的博客
08-24 2585
由于项目需要使用Jetson nano开发板，所以想在自己电脑上安装ubuntu18.04并部署TensorRT加速的yolov5，这样一来所有的调试都可以在本地做好，不然的话板子性能较低，跑什么都慢，浪费时间。 一、选择兼容的版本 在开始之前，最最重要的就是版本的选择，想在ubuntu上让TensorRT的yolov5跑起来，中间涉及到非常多的版本选择，一旦不配套，就会直接面临重装，相当麻烦。 >首先列出需要作出版本选择的条目，以及我成功时使用的版本 ubuntu kernel
yolov5检测代码解析
热门推荐
weixin_41251668的博客
09-08 1万+
yolov5检测部分代码解析 设置目标检测的配置参数： parser = argparse.ArgumentParser() parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pt', help='model.pt path(s)') parser.add_argument('--source', type=str, default='data/images', help='source') # f
jetbot11 之 人手目标检测(hand detect)
狮山飞虎的的博客
12-11 4620
一直陷在物体检测的坑里出不来了。谁让这坑如此之深 ！ 继续加点深度 ！ 自己训练一个 object detect 模型 再跑跑 tensorRT 加速吧。 技术主题： yolov5s 训练人手检测模型并使用 tensortRT 加速。 一 准备数据集 1.1 下载 https://www.robots.ox.ac.uk/~vgg/data/hands/ 1.2 转换 yolo 需要的格式 参考： https://github.com/coldlarry/YOLOv3-complete-.
tensorrt安装
weixin_38906715的博客
09-18 321
The following packages have unmet dependencies:tensorrt : Depends: libnvinfer8 (= 8.0.0-1+cuda10.2) but 8.0.0-1+cuda11.3 is to be installeddpkg -i nv-tensorrt-repo-ubuntu1604-cuda10.2-trt8.0.1.6-ga-20...
ubuntu18.04系统部署yolov5
05-31
好的，我可以为您提供一些指导。以下是在Ubuntu 18.04上部署Yolov5的步骤： 1. 首先，确保您的系统已经安装了CUDA和cuDNN。您可以通过以下命令检查CUDA的版本： ``` nvcc --version ``` 您可以通过以下命令检查cuDNN的版本： ``` cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2 ``` 2. 然后，安装Python3和pip3： ``` sudo apt-get update sudo apt-get install python3-dev python3-pip ``` 3. 安装Yolov5所需的Python库： ``` pip3 install numpy torch torchvision opencv-python ``` 4. 克隆Yolov5仓库： ``` git clone https://github.com/ultralytics/yolov5.git ``` 5. 进入仓库并下载预训练权重： ``` cd yolov5/ wget https://github.com/ultralytics/yolov5/releases/download/v3.1/yolov5s.pt ``` 6. 接下来，您可以使用Yolov5进行推理或训练。如果您要使用预训练权重进行推理，请使用以下命令： ``` python3 detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source 0 ``` 这将在摄像头视频流上运行Yolov5，并将检测结果输出到屏幕上。 以上就是在Ubuntu 18.04上部署Yolov5的步骤。希望对您有所帮助！
评论 7
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值