୧⍤⃝ ୧⍤⃝ 2023-04-21 01:28 采纳率: 51.3%
浏览 53
已结题

YOLOv5使用GPU训练以后无法对目标图片进行识别

YOLO v5 使用GPU训练完成以后对目标图片进行识别显示以下代码，请问是哪里出问题了

img

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2023-04-21 09:39
    关注

    oom错误，显存不足，将imgs这个参数改小，你这8032的的图片没几张显卡跑的了的

    img

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(1条)

报告相同问题？

  • YOLO v5训练使用CPU正常，使用GPU训练 box cls obj均为nan pytorch 深度学习 目标检测
    2022-01-15 21:21
    回答 2 已采纳 解决了，既然锁定问题出在显卡上，在坚信不是硬件问题的前提下，只能一遍一遍换驱动版本。 刚开始用的最早的驱动版本是11.3，pytorch支持的另一个cuda版本10.2 nvidia官网不支持wind
  • YOLO V5怎么多GPU同时训练一个项目？ 深度学习 神经网络
    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
  • YOLOv5+DeepSORT对视频进行目标检测追踪计数速度慢 opencv python
    2022-11-12 10:49
    回答 2 已采纳 以下几个优化方向，希望对你有帮助 视频的帧率你可以适当调低，比如以前60帧，你可以变成25帧，30针的样子处理就行了，通过代码进行帧判断预测时不要逐帧进行。视频的分辨率也可以调低。更换轻量的模型
  • yolov5目标识别检测.rar
    2021-11-22 21:48
    总的来说，这个项目提供了一个使用YOLOv5和OpenCV进行身份证检测的实例，对于学习目标检测技术，特别是身份证识别，是一个非常实用的起点。通过理解和研究源代码，开发者可以深入理解YOLOv5的工作原理，以及如何在...
  • yolov5训练模型时遇到nan 人工智能 机器学习 算法
    2022-12-31 20:12
    回答 2 已采纳 很遗憾，我无法确定你遇到的问题的原因。在没有更多的信息的情况下，我只能给出一些潜在的建议。 首先，如果你发现模型使用更大的权重文件时在 GPU 上运行会导致数据全部变为 nan，可能是因为你的 GP
  • yolov5训练结果只有loss为0 python 机器学习 深度学习
    2022-10-13 20:50
    回答 1 已采纳 单个类别的时候，train里面要将single-cls设置为true，或者参数里面加上--single-cls
  • yolov5为什么默认要训练300次这么多 人工智能 深度学习 神经网络
    2021-04-05 20:56
    回答 2 已采纳 mmdetection没试过，但是yolov5默认300多次是因为人家是在coco数据集上面训练的，而你看yolov5的主页，里面的各个模型的对比可以看出来，基本上都是300epoch的时候达到较好的
  • yolov8 模型训练GPU环境配置保姆级教程
    2024-06-23 11:30
    Cc不爱吃洋葱的博客 这些大型预训练模型，如GPT-3、BERT、XLNet等，以其强大的语言理解和生成能力，正在改变我们对人工智能的认识。无论您是科研人员、工程师，还是对AI大模型感兴趣的爱好者，这套报告合集都将为您提供宝贵的信息和启示...
  • Yolofastest_v2 使用GPU训练速度很快，识别速度很慢？ python 深度学习 计算机视觉
    2023-01-28 21:40
    回答 2 已采纳 跑的时候看下开下任务管理器，这里切换成cuda，然后运行看下这里的cuda和下面的现存有无上升，有上升说明有调用显卡。另外，你的模型有无预热？一般来说，显卡推理的前一两张速度会很慢，后面速度就飞起了。
  • yolov5模型训练问题 pytorch 机器学习 目标检测
    2022-10-11 21:09
    回答 3 已采纳 你的pytorch的cuda版本选错了，torch1.7.0是支持cuda11.x的，但是你下载的是cu101,这个就表示支持的cuda是10.1。但是你环境里面的cuda是11.x（30系显卡不支持
  • yolov5训练voc2007 python
    2022-08-05 17:33
    回答 2 已采纳 最后通过朋友帮助，发现batch_size的大小会影响到训练的收敛效果。解决方法：batch_siz最初设置为4改为batch_siz=8就可以了
  • yolov5_FaceMask：检测带有或不带有口罩的人。 使用YOLOv5训练
    2021-02-06 08:35
    YOLOv5_FaceMask项目是一个利用深度学习技术进行面部口罩检测的应用，旨在识别出图像中的人是否佩戴了口罩，特别是在COVID-19大流行期间，这种技术对于公共卫生监测和安全有着重要的作用。该项目基于YOLOv5模型，这...
  • yolov5 解码使用GPU进行加速
    2021-11-23 19:30
    zsffuture的博客 1.yolov5系列的原始输出是3个head头，上图画的是输入为608*608的分辨率的图，如果输入改为640*640分辨率的图片，那么输出的3个头分别对应三个8、16、32下采样的输出分别为80*80*255、40*40*255、20*20*255，其中对应...
  • yolov5 deepsort 车辆识别计数完整论文
    2022-04-20 15:18
    1. **基于YOLOv5的车辆目标检测**：首先利用YOLOv5模型对采集到的交通视频进行车辆检测，精简模型至只检测车辆类别，提高检测的速度和精度。 2. **DeepSORT多目标跟踪**：结合YOLOv5输出的结果，采用DeepSORT算法...
  • C# OpenCvSharp DNN 部署yolov5旋转目标检测 源码
    2023-12-13 15:04
    在本文中，我们将深入探讨如何使用C#编程语言结合OpenCvSharp库和Darknet的预训练模型Yolov5实现旋转目标检测。这个项目的关键在于利用深度神经网络（DNN）技术，使得我们能够在图像中准确地检测并识别出各种角度的...
  • 没有解决我的问题, 去提问

问题事件

  • 系统已结题 4月29日
  • 已采纳回答 4月21日
  • 创建了问题 4月21日