正文

Yolov5 物体检测训练

 2023-03-12  87

关键词：

【中文标题】Yolov5 物体检测训练【英文标题】：Yolov5 object detection training 【发布时间】：2022-01-08 22:54:35 【问题描述】：

关于我的目标检测yolov5 训练过程，我需要您的帮助！

我尝试训练我的物体检测模型 yolov5 来检测小物体（从头开始）。为了标记我的图像，我使用了 roboflow，我应用了一些数据增强和一些 roboflow 作为服务提供的预处理。当我完成预处理步骤并且数据增强 roboflow 提供了不同输出格式的选择时，在我的情况下它是 yolov5 pytorch，并且 roboflow 为我将数据拆分为训练验证和测试。因此，一切都按照我的数据准备进行了设置，最后我得到了带有 data.yaml 的文件夹和带有标签的图像，在 data.yaml 中，我把我看到的训练和验证集的路径在 yolov5 的 GitHub 教程中。 我非常仔细地按照步骤进行操作。

问题是当训练开始时，我在 obj 和 box 列中得到 nan，如下图所示，我不知道原因，有人能与此相关吗或者请给我任何线索以找到解决方案，这是我在计算机视觉领域的第一个项目。

This is what i get when the training process starts

This the last message error when the training finish

I think the problem comes maybe from here but i don't know how to fix it, i used the code of yolov5 team as it's in the tuto

训练继续进行，没有任何问题，但地图和精度在整个过程中保持为 0！！

Ps：这是我关注的tuto链接：https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data

【问题讨论】：

这就是我要解决的问题。 - 在 collab 上运行您的代码，因为环境被证明运行良好 - 确认您的标签看起来不错并且设置正确。您可以检查以确保课程看起来正确吗？在其中一张屏幕截图中，您似乎没有标签。非常感谢@alexheat 的回答！我将按照您的建议在 colab 中测试我的代码，并让您知道它是否工作正常。关于标签，我检查了他们的设置并进行了一些可视化，显示盒子应该是这样并且看起来很好，对于我只有一个类的类，该类的标签是 0，因为 yolov5 从一个标签的 0 开始，如果有第二个标签，则第二个标签为 1。我不知道这是否与我的问题有关（在我的模型代码实现中的某个地方除以 0 作为示例，因为我自己没有编写代码，所以我采用了 yolov5 的代码，因为它是如此我不知道代码里面的步骤和操作）。 【参考方案1】：

这就是我要解决的问题。 - 在 collab 上运行您的代码，因为环境被证明运行良好 - 确认您的标签看起来不错并且设置正确。您可以检查以确保课程看起来正确吗？在其中一张屏幕截图中，您似乎没有标签

【讨论】：

【参考方案2】：

在 colab 中运行我的代码成功，结果很好。我认为问题出在我的个人笔记本电脑环境中，可能是我使用的 pytorch 版本“1.10.0+cu113”，或者其他什么！如果您有任何建议可以正确地为 yolov5 设置我的环境，我很乐意向你们学习。非常感谢@alexheat

【讨论】：

欢迎您。如果你喜欢我的回答，请接受我的回答。我自己一直在 Google Colab 上做，因为我自己的笔记本电脑也无法使用它 done (y) 我搜索了一下，发现问题可能来自于 Torch 的版本，因为在我之前提到的笔记本电脑中，我有 1.10.0+cu113 但在教程中yolov5，他们使用的是 1.10.0+cu111，也许你可以试试这个版本，看看它是否有效 :) 我自己还没有尝试过，但如果我这样做了，我会告诉你结果！

使用ymir生产基于yolov5的头盔检测模型

使用YMIR生产基于yolov5的头盔检测模型1.概述2.YOLOV5结构解析YOLOV5在coco数据集性能测试图3.算法基本信息动手实测查看训练、测试数据集模型训练启动页面模型运行中性能、进度曲线变化启动训练任务后即可查看训练过程模型发布... 查看详情

目标检测及目标定位(代码片段)

...体在图像中的位置来分别判断的情况，而且对应的是YOLOv5模型。YOLOv5目标检测的内容可以看看我之前的一篇文章YOLOv5训练自己的数据集_ONEPIECE_00的博客-CSDN博客本文采用的目标定位的方法，其实就是根据物体检测后得到的... 查看详情

深度学习目标检测：yolov5实现车辆检测(含车辆检测数据集+训练代码)(代码片段)

深度学习目标检测：YOLOv5实现车辆检测(含车辆检测数据集+训练代码) 目录YOLOv5实现车辆检测(含车辆检测数据集+训练代码)1.前言2.车辆检测数据集说明（1）车辆检测数据集（2）自定义数据集3.基于YOLOv5... 查看详情

深度学习目标检测：yolov5实现车辆检测(含车辆检测数据集+训练代码)(代码片段)

快速入门yolov5目标检测算法(代码片段)

文章目录一、YOLOv5简介二、网络结构1、Input2、Backbone3、Neck4、Head三、改进方法1、自适应锚框计算2、自适应灰度填充四、性能表现五、YOLOv5入门实战一、YOLOv5简介YOLOv5是一个在COCO数据集上预训练的物体检测架构和模型系列，... 查看详情

行人检测(人体检测)2：yolov5实现人体检测(含人体检测数据集和训练代码)(代码片段)

行人检测(人体检测)2：YOLOv5实现人体检测(含人体检测数据集和训练代码)目录行人检测(人体检测)2：YOLOv5实现人体检测(含人体检测数据集和训练代码)1.前言2.人体检测数据集说明（1）人体检测数据集（2）... 查看详情

智能驾驶车牌检测和识别《yolov5实现车牌检测（含车牌检测数据集和训练代码）》(代码片段)

智能驾驶车牌检测和识别（二）《YOLOv5实现车牌检测（含车牌检测数据集和训练代码）》目录智能驾驶车牌检测和识别（二）《YOLOv5实现车牌检测（含车牌检测数据集和训练代码）》1.前言2.车牌检... 查看详情

yolov5训练coco128数据集和测试与检测(代码片段)

yolov5训练coco128数据集和测试与检测文章目录yolov5训练coco128数据集和测试与检测参考链接一、coco数据集1.简介2.下载3.解压后的数据4.COCO数据集(.json)训练格式转换成YOLO格式(.txt)二、训练train.py三、验证val.py四、推理预测detect.py1.用... 查看详情

yolov5训练coco128数据集和测试与检测(代码片段)

智能驾驶车牌检测和识别《yolov5实现车牌检测（含车牌检测数据集和训练代码）》(代码片段)

深度学习目标检测：yolov5实现红绿灯检测(含红绿灯数据集+训练代码)(代码片段)

深度学习目标检测：YOLOv5实现红绿灯检测(含红绿灯数据集+训练代码)1.前言本篇博客，我们将手把手教你搭建一个基于YOLOv5的红绿灯目标检测项目。目前，基于YOLOv5s的红绿灯检测精度平均值mAP_0.5=0.93919，mAP_0... 查看详情

深度学习目标检测：yolov5实现红绿灯检测(含红绿灯数据集+训练代码)(代码片段)

笔记2：yolov5训练自己的目标检测模型_创建并划分数据集(代码片段)

...检测的数据集资源标签的格式都是VOC(xml格式)的，而yolov5训练所需要的文件格式是yolo(txt格式)的，这里就需要对xml格式的标签文件转换为txt文件。同时训练自己的yolov5检测模型的时候，数据集需要划分为训练集和验证查看详情

用yolov5ds训练自己的数据集——同时检测和分割(代码片段)

...常感谢作者midasklr的开源项目！源码地址：midasklr/yolov5ds:multi-taskyolov5withdetectionandsegmentation(github.com)https://github.com/midasklr/yolov5ds关于作者：CSDN：MidasKing的博客_CSDN博客-目标检测,python,opencv领域博主GitHub：midasklr(g... 查看详情

学习对象检测检测结果显示变色

...：简要说明最近开始学习物体检测，刚开始学习PyTorch，YOLOv5。所以我想为什么不建一个小项目来学习呢？用它来训练检测皮卡丘。问题我已经成功地用皮卡丘训练了模型，然后使用训练过的权重和我自己编写的Python脚查看详情

yolov5实现佩戴安全帽检测和识别(含佩戴安全帽数据集+训练代码)(代码片段)

YOLOv5实现佩戴安全帽检测和识别(含佩戴安全帽数据集+训练代码)目录YOLOv5实现佩戴安全帽检测和识别(含佩戴安全帽数据集+训练代码)1.前言2.佩戴安全帽检测和识别的方法（1）基于目标检测的佩戴安全帽识别方法... 查看详情

yolov5训练&检测流程(代码片段)

文章目录训练流程第一步：将训练所用数据分别放入`Annotations`、`images`文件夹当中当中第二步：删除以下文件，**仅删文件请勿删除文件夹**第三步：依次运行`split_train_val.py`、`voc_label.py`文... 查看详情