YOLO v5 使用GPU训练完成以后对目标图片进行识别显示以下代码，请问是哪里出问题了
YOLOv5使用GPU训练以后无法对目标图片进行识别
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
分享- 邀请回答
-
2条回答 默认 最新
爱晚乏客游 2023-04-21 09:39关注oom错误，显存不足，将imgs这个参数改小，你这8032的的图片没几张显卡跑的了的
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏举报 编辑记录 分享
- 2023-04-21 01:28回答 2 已采纳 oom错误，显存不足，将imgs这个参数改小，你这8032的的图片没几张显卡跑的了的
- 2021-05-11 10:28回答 1 已采纳 parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') defau
- 2022-01-15 21:21回答 2 已采纳 解决了，既然锁定问题出在显卡上，在坚信不是硬件问题的前提下，只能一遍一遍换驱动版本。 刚开始用的最早的驱动版本是11.3，pytorch支持的另一个cuda版本10.2 nvidia官网不支持wind
- 2021-11-23 19:30zsffuture的博客 1.yolov5系列的原始输出是3个head头，上图画的是输入为608*608的分辨率的图，如果输入改为640*640分辨率的图片，那么输出的3个头分别对应三个8、16、32下采样的输出分别为80*80*255、40*40*255、20*20*255，其中对应...
- 2022-11-12 10:49回答 2 已采纳 以下几个优化方向，希望对你有帮助 视频的帧率你可以适当调低，比如以前60帧，你可以变成25帧，30针的样子处理就行了，通过代码进行帧判断预测时不要逐帧进行。视频的分辨率也可以调低。更换轻量的模型
- 2021-04-05 20:56回答 2 已采纳 mmdetection没试过，但是yolov5默认300多次是因为人家是在coco数据集上面训练的，而你看yolov5的主页，里面的各个模型的对比可以看出来，基本上都是300epoch的时候达到较好的
- 2022-08-05 17:33回答 2 已采纳 最后通过朋友帮助，发现batch_size的大小会影响到训练的收敛效果。解决方法：batch_siz最初设置为4改为batch_siz=8就可以了
- 2022-05-24 06:30十年一梦实验室的博客 Custom Object Detection Training using YOLOv5 | LearnOpenCV使用 YOLOv5 进行自定义目标检测训练Table of Contents1.Introduction2.What is YOLOv5?1.Models Available in YOLOv52.Features Provided by YOLOv53....
- 2022-12-31 20:12回答 2 已采纳 很遗憾，我无法确定你遇到的问题的原因。在没有更多的信息的情况下，我只能给出一些潜在的建议。 首先，如果你发现模型使用更大的权重文件时在 GPU 上运行会导致数据全部变为 nan，可能是因为你的 GP
- 2023-01-28 21:40回答 2 已采纳 跑的时候看下开下任务管理器，这里切换成cuda，然后运行看下这里的cuda和下面的现存有无上升，有上升说明有调用显卡。另外，你的模型有无预热？一般来说，显卡推理的前一两张速度会很慢，后面速度就飞起了。
- 2022-10-11 21:09回答 3 已采纳 你的pytorch的cuda版本选错了，torch1.7.0是支持cuda11.x的，但是你下载的是cu101,这个就表示支持的cuda是10.1。但是你环境里面的cuda是11.x（30系显卡不支持
- 2024-03-14 23:52知来者逆的博客 从Yolov9的算法解读和之前模型对比到详细记录模型训练与推理的过程
- 2022-05-30 10:50回答 2 已采纳 yolov5必须相同算力的gpu才能够运行，算力不同，模型文件需要重新生成。gpu算力查询 https://developer.nvidia.com/cuda-gpus#compute
- 2021-08-20 15:19野生蜘蛛侠的博客 yolov5的配置相对简单，但是网上大部分博客都没有仔细介绍具体步骤，本文将详细介绍YOLOv5在windows10系统下的环境配置的全过程，以及yolov5使用的一些细节，以及如何制作和训练自己的数据集。 一.搭建环境 1.本...
- 2020-07-30 23:58我爱计算机视觉的博客 本文会详细介绍YOLO V5的网络结构及组成模块，并使用YOLO V5s在BDD100K自动驾驶数据集上进行迁移学习，搭建属于自己的自动驾驶交通物体对象识别网络。本文来源：知乎-自动驾驶...
- 没有解决我的问题, 去提问
问题事件
系统已结题
4月29日
已采纳回答
4月21日-
创建了问题
4月21日
悬赏问题
- ¥20 怎么用dlib库的算法识别小麦病虫害
- ¥15 华为ensp模拟器中S5700交换机在配置过程中老是反复重启
- ¥15 java写代码遇到问题，求帮助
- ¥15 uniapp uview http 如何实现统一的请求异常信息提示？
- ¥15 有了解d3和topogram.js库的吗？有偿请教
- ¥100 任意维数的K均值聚类
- ¥15 stamps做sbas-insar，时序沉降图怎么画
- ¥15 买了个传感器，根据商家发的代码和步骤使用但是代码报错了不会改，有没有人可以看看
- ¥15 关于#Java#的问题，如何解决？
- ¥15 加热介质是液体，换热器壳侧导热系数和总的导热系数怎么算