YOLOv5-v3.1,推理环境配置、Tensorrt加速一步到位(各种问题总结,吐血整理)

本文详细介绍了如何配置YOLOv5-v3.1的TensorRT推理环境,包括opencv的安装,解决安装过程中遇到的问题,以及TensorRT的下载、安装和验证。此外,还分享了在CUDA 11.0和PyTorch 1.10环境下,使用TensorRT 7.2.3.4加速YOLOv5模型时的关键步骤和常见问题的解决方案。

cuda11.0 ,pytorch 1.10 ,tensorrt7.2.3.4 ,其中会需要opencv c++ 安装,和tensorrt安装


一、opencv安装

1.下载opencv

https://opencv.org/releases/
在这里插入图片描述
sources 或者 GitHub都可以,我用的Sources
下载后解压zip 文件

unzip opencv-4.5.3.zip

2.安装环境

需要的指令为:

sudo apt-get install build-essential
sudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev 
sudo apt-get install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev

主要是第一,第二条。第三条可以不需要。

可能存在的问题1unable to locate
在这里插入图片描述
解决方法

sudo apt-get update

可能存在的问题2:无法定位 libjasper-dev

sudo add-apt-repository “deb http://security.ubuntu.com/ubuntu xenial-security main”
sudo apt update
sudo apt install libjasper1 libjasper-dev

可能存在的问题3:command not found
在这里插入图片描述
解决方法

sudo apt-get install software-properties-common

3.正式安装

(1)看你opencv位置在哪, cd到哪里,我是放在/home下,opencv文件夹下创建build文件夹

cd opencv
mkdir build
cd build

YOLOlayer.h是一个头文件,其中定义了YOLO层的相关函数和结构体。该头文件包含了以下内容: 1. 构造YOLO层的函数make_yolo_layer,用于创建YOLO层的实例。 2. YOLO层的前向传播函数forward_yolo_layer,用于执行YOLO层的前向传播操作。 3. YOLO层的反向传播函数backward_yolo_layer,用于执行YOLO层的反向传播操作。 4. 调整YOLO层大小的函数resize_yolo_layer,用于调整YOLO层的输入尺寸。 5. 计算YOLO层检测结果数量的函数yolo_num_detections,用于计算YOLO层的检测结果数量。 此外,该头文件还包含了一些与GPU相关的函数,如forward_yolo_layer_gpu和backward_yolo_layer_gpu,用于在GPU上执行YOLO层的前向传播和反向传播操作。\[1\] YOLOlayer.h是darknet源码中的一个文件,主要用于实现YOLO v3中的三个detection层,分别对应52*52*75,26*26*7513*13*75的输出。这些层是YOLO v3论文的核心部分。\[2\] 此外,YOLOlayer.h还与TensorRT加速YOLOv5相关。可以通过GitHub手动获取对应版本的tensorrtx,也可以使用相应版本的指令进行安装。\[3\] #### 引用[.reference_title] - *1* *2* [【darknet源码解析-20】yolo_layer.h 和 yolo_layer.c 源码解析](https://blog.csdn.net/caicaiatnbu/article/details/102962445)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [YOLOv5-v3.1推理环境配置Tensorrt加速一步到位(各种问题总结,吐血整理)](https://blog.csdn.net/knowledge112233/article/details/126262744)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论 5
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值