YOLO过程

一、YOLOV1

在这里插入图片描述yolov1最后输出的是7x7x30,即最后的特征层为7x7=49个像素,因为是下采样后的,所以每个像素对应原图中的一块区域
而30=1+4+1+4+20,也就是每个网格有2个bbox,输出的是位置信息和置信度,以及20分类

YOLOV1分为训练阶段、预测阶段

1.训练阶段

训练阶段,每个像素点(对应原图中的一块区域)预测两个Bbox,假设图像中有两个物体,原图中待测对象的中心点落在哪个网格中,则就由这个网格中的两个bbox进行预测对象,两个bbox又根据置信度舍去一个;
但是在这还要注意一个问题:其他grid预测的bbox咋办那,因为有置信度公式,它们也会计算置信度,只需要做到让它们的置信度非常小就行。
每个网格也只能预测一个类别
训练阶段中也就产生了三个损失:置信度、位置、种类损失;
最后网络也就输出了预测框的位置(x,y,w,h,conf,c);
通过损失函数反向传播------得到训练后的权重

置信度= bbox与GT的IOU值

2.预测阶段

加载训练之后的权重后,网络最后输出的仍然是7x7x30的张量,最后有49*2=98个bbox;
①首先通过设置confidence的阈值,去除掉一部分bbox,有的bbox中可能没有物体
②假如是20分类,然后每个类别都会进行NMS处理,选择最优的框,这样就得到了某个种类的预测框
NMS处理:20个种类,假设第一个种类是dog,将经过第一步骤后的bbox,根据是dog的概率进行排序,最大的先作为样本,然后依次给他作IOU比较,【所以,IOU设置过小,那么去除的框也就非常多,只有超过了这个阈值,才会去除掉剩余的框】

3.YOLOv1不足

1.对于群体的小目标检测效果差,因为每个网格的两个bbox都是针对同一个类别的置信度,所以比如当一群鸟飞过的时候,检测效果差;
2.不是基于anchor
3.对新的目标纵横比检测效果差

====================================================================================================================================================================================================================

二、YOLOV2

yolov1直接最后输出的是(7x7x30)的特征,直接输出bbox;
Fastet R-CNN、YOLOv2等以后算法都是基于anchor

1.什么是anchor

在深度学习时代,即使是R-CNN、Fast R-CNN都是基于滑窗来产生候选框,直到Faster R-CNN提出了RPN网络,RPN网络一个重要的概念就是anchor

窗口滑动的时候,本质就是遍历像素的过程,因此直接为每个像素分配不同的尺度和比例的窗口矩形,它们的中心都是其所属的像素点。对于长度和比例的分配们可以根据标注图像信息通过k-means聚类得到。而每个像素分配几个不同长度和比例的窗口举行就是Anchor。

2.举例

假设一张256256大小的图片,经过64,、128、256倍率的下采样,产生44,、22、11大小的特征图,在这三个特征图每个像素点上设置三个不同大小的矩形框,就是anchor。所以来说anchor的位置是固定的,从而根据偏移、倍率计算出真实的预测框。

在这里插入图片描述

3.anchor大小、比例设置也很重要

在这里插入图片描述
如图,电线杆的宽高比可能不止1:10,而且特别细。如果您设置的anchor长宽比为1:1、1:2、 2:1、 1:3、 3:1这五个不同的长宽比,那可能到导致在训练的时候,没有哪个anchor与电线杆的IoU大于0.5,导致全部为负样本。
在YOLOV3中9个anchor的大小如下,(有三个特征层,每个特征层上每个像素点有三个anchor)
在这里插入图片描述

三、YOLOV3

在这里插入图片描述

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
YOLO(You Only Look Once)是一种实时目标检测算法,它能够在图像或视频中快速准确地检测出多个目标。下面是YOLO识别的开发过程: 1. 数据收集和标注:首先需要收集大量的图像和视频数据,并对其中的目标进行标注,标注包括目标的位置和类别信息。 2. 数据预处理:对收集到的数据进行预处理,包括图像的缩放、裁剪、增强等操作,以提高模型的泛化能力和鲁棒性。 3. 构建网络模型:使用深度学习框架(如TensorFlow、PyTorch)构建YOLO模型。YOLO模型通常由卷积层、池化层、全连接层等组成,其中卷积层用于提取图像特征,全连接层用于预测目标的位置和类别。 4. 模型训练:使用标注好的数据对YOLO模型进行训练。训练过程中,通过反向传播算法不断调整模型参数,使得模型能够更准确地预测目标的位置和类别。 5. 模型优化:对训练好的模型进行优化,包括调整超参数、使用正则化技术、增加数据量等方法,以提高模型的性能和泛化能力。 6. 模型评估:使用测试数据对训练好的模型进行评估,计算模型的准确率、召回率、精确率等指标,以评估模型的性能。 7. 部署和应用:将训练好的模型部署到实际应用中,可以通过调用模型的API接口实现目标检测功能。在应用中,可以实时输入图像或视频数据,通过YOLO模型进行目标检测,并输出检测结果。

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

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值