Pytorch 2.0 Yolov5推理性能提升记录

炼丹去了

已于 2023-05-08 19:22:05 修改

阅读量427

点赞数

文章标签： pytorch YOLO ubuntu

于 2023-05-08 13:48:34 首次发布

本文链接：https://blog.csdn.net/djj199301111/article/details/130557339

版权

测试环境

Nvidia A30
CUDA 11.7
Python 3.9
Pytorch 2.0
Ubuntu 20.04
Yolov5 v7.0

测试结果记录：

模型	size	compile	Speed（ms/image）	显存占用（GB）
yolov5m	640	×	13.4	1.7
yolov5m	640	√	11.3	1.6
yolov5m	640	×	16.2	2.2
yolov5m	640	√	14.6	2.1

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

炼丹去了

关注关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
1
评论
Pytorch 2.0 Yolov5推理性能提升记录

测试环境Nvidia A30CUDA 11.7Python 3.9。
复制链接

扫一扫

YOLOV5（一）：win10+pytorch配置yolov5网络训练数据集

abandononeself的博客

07-24

325

win10+pytorch配置yolov5网络训练数据集

PyTorch+Yolov5环境搭建

weixin_44332602的博客

10-11

1680

免额外安装cuda、cudnn

1 条评论您还未登录，请先登录后发表或查看评论

PyTorch 2.0 实操，模型训练提速！

datayx的文章

02-03

1908

向AI转型的程序员都关注了这个号????????????PyTorch 2.0 官宣了一个重要特性 —— torch.compile，这一特性将 PyTorch 的性能推向了新的高度，并将 PyTorch 的部分内容从 C++ 移回 Python。torch.compile 是一个完全附加的（可选的）特性，因此 PyTorch 2.0 是 100% 向后兼容的。支撑 torch.compile 的技术包括研发团队新...

（linux）yolov5/v3-pytorch目标检测2：训练自己的数据集，gpu，pycharm, RTX2080ti，单/多显卡训练，笔记

chao_xy的博客

03-11

859

（linux）yolov5/v3-pytorch目标检测2：训练自己的数据集，gpu，pycharm, RTX2080ti，单/多显卡训练。yolov3和yolov5可以使用相同环境配置，数据集和yaml文件通用，训练步骤相同。

【深度学习】1. yolov5 推理速度和batchsize的增长关系，推理并行处理多张图片，显存如何手动释放

weixin_40293999的博客

07-05

5060

yolov5的detect.py 是默认batchsize=1的，而直接修改batchsize为更大的值，会使detect.py 不工作.因此本篇文章想解决两个问题：1.设置多大的bz合适2.如何修改detect.py 使其满足自定义的bz。

pytorch导出yolov5 onnx模型用vs2019 c++推理源代码

07-07

本代码是根据yolov5导出的onnx模型，通过c++部署；本代码是可运行代码，但是项目目录下的包含目录需要...如果调试过程中有其他疑问，可以先看看我的这篇博客：pytorch导出yolov5 onnx模型用vs2019 c++推理保姆级教程

pytorch_yolov5_weights.rar

04-19

yolov5官方提供的pt模型，网络不好时下载较慢，这里将下好的模型分享给大家。已经验证过yolov5s的pt模型可以通过tensortx中的方法，转换为TensorRT可以的格式，并跑通流程。

pytorch搭建YOLOV5模型

11-13

YOLOv5 是一种基于深度学习框架 PyTorch 的目标检测算法，其核心技术是一种轻量级的目标检测模型。它通过将输入图像分成不同大小的网格，每个网格负责检测该位置是否存在目标，并预测目标的位置和类别。与传统的目标...

Pytorch实现yolov5

08-19

使用Pytorch实现的yolov5目标检测算法，可以训练自定义的数据集，包含多个主干网络cspdarknet，mobilenet等可以选择，支持CPU或者GPU，可以用来检测图片或者视频，训练的时候支持迁移学习。

基于pytorch的Yolov5人脸检测

最新发布

02-07

基于pytorch的Yolov5人脸检测基于pytorch的Yolov5人脸检测

基于 PyTorch实现YOLOv5

ssl22020827的博客

09-21

5455

本文记录基于PyTorch实现Github作者ultralytics的YOLOv5-tag v5.0的迁移学习大致流程，使用自制数据集（100张图片）。

yolov5_obb报错合集

qq_44783689的博客

03-29

1868

yolov5 obb的cuda版本和pytorch版本最好和给的一样，pytorch版本不能超过1.10，这样就能减少很多报错，所以强烈建议配置官方教程的版本并且要求：CUDA Driver Version ≥ CUDA Toolkit Version(runtime version) = torch.version.cuda然而，我的cuda是11.7和pytorch为2.0，所以调通前问题一个接一个。下面是我遇到的部分问题。

yolov5环境搭建与pytorch中torch、torchvision、torchaudio安装

zhang1112222的博客

11-07

3794

pytorch

YOLO项目服务器配置及云硬盘挂载问题

pengxiang1998的博客

02-02

1493

依次执行就可以，本次博主又换了一个服务器，这次在安装过程中遇到了一些问题，最终也解决了，实验数据集换成了CADCD，输入尺寸为1280x1024，很明显此时的显存占用要比640x640的大了很多，下面是本次GPU使用情况：batch-size为16，迭代次数为200次。在使用这个服务器的过程中意识到每次更换服务器需要重新上传项目和数据集，这让人很苦恼，因此可以采用加装一个云硬盘的方式，这样就不需如此频繁更换了，然而也是刚刚意识到该问题，博主的GPU服务器是硅谷二区的，但并未找到可用区的硬盘，呜呜呜。

YOLOV5与YOVOV7，YOVOV7并未达到预期效果，求解

qq952936952的博客

11-09

3579

YOLOV5 同情况下节省一半以上，300张图像 batch-size 16 占显存 8G 左右速度每轮次10秒。YOLOV7 300张图像 batch-size 只能到 8 往上就显存不足显存占满速度每轮次40秒。检测速度，检测精度，实际情况下都不是很理想，需要调整那些的参数。YOLOV7 目测百分之10的误判或漏检科涅还不止。YOLOV5 目测百分之一的误判或漏检。有没有大神知道为什么会出现类似的效果，1，训练所需要资源差别。

【机器学习】全网最全模型评价指标（性能指标、YOLOv5训练结果分析、轻量化指标、混淆矩阵详解）【基础收藏】

热门推荐

深度学习

03-27

1万+

【机器学习】全网最全模型评价指标（性能指标、YOLOv5训练结果分析、轻量化指标、混淆矩阵详解）【基础收藏】

关于Win10 环境下 YOLO3 GPU 版配置的一些问题（都是我踩过的坑）

魔法学徒

03-08

6543

教程不写了，我是参考这个链接的 win10下YOLOv3下载与编译严格按照教程，提醒要注意的点和配置中遇到的问题环境 Win10 CUDA 9.0 + CUDNN 9.0 Visual Studio 2017 OpenCV 3.4.0 YOLO v3 提醒如下： cuda 版本要与自己显卡版本对应，且 &amp;gt;= 7.5 VS 版本 2013 以上VS 版本 2013 以上 OpenC...

Yolov5 计算访存量MAC与计算量FLOPS

m0_46093829的博客

12-02

3338

Yolov5 计算访存量MAC与计算量FLOPS

YOLOV5 配 pytorch 环境

weixin_38643701的博客

03-20

2131

YOLOV5 配 pytorch 环境 1.安装pytorch (1)安装Anaconda 1)先去官方地址下载好对应的安装包下载地址： https://www.anaconda.com/download/#linux 2)然后安装anaconda bash ~/Downloads/Anaconda3-2020.11-Linux-x86_64.sh 需要添加环境变量。编辑~/.bashrc 文件 vi ~/.bashrc 在最后一行加上 export PATH=/h..

pytorch在yolov5算法哪里体现了

05-24

2. 训练与推理：PyTorch提供了灵活的训练和推理接口，可以方便地对YoloV5模型进行训练和推理，同时也提供了GPU加速，可以大大提升训练和推理的效率。 3. 数据加载与处理：YoloV5算法需要大量的训练数据来完成目标...

“相关推荐”对你有帮助么？

非常没帮助
没帮助
一般
有帮助
非常有帮助

提交