图解目标检测算法YOLOv1,保姆级教程。

2021年5月31日08:53:10 8 7,525 °C
摘要

目标检测算法,YOLOv1的超详细解析,一文看懂YOLO。

大家好,我是 Jack。

承诺的图解 AI 算法系列教程,今天咱们继续

这个系列一直写的比较随性,想写哪个算法就写了哪个,毫无章法。

「修炼开始」一文带你入门深度学习

保姆级教程:硬核图解Transformer

嘿,来聚个类!

好在,文章质量都还不错,虽然硬核了点,但从各方面的反馈来看,还是有不少朋友喜欢看的。

今天,聊一聊人工智能,计算机视觉方向的重头戏

我们都知道,CV 领域最常规的三大任务是:图像分类、目标检测、图像分割。

图像的分类和分割算法实战教程,我在2019年就出过了,想看可以往前翻一翻。

目前,就差目标检测还没写过,虽迟但到,今天它来了

目标检测

目标检测,应用非常广,是非常重要的技术储备。

目标检测落地场景丰富,可用度非常高,人脸识别、智能安防系统、无人驾驶、无人机等各个领域,目标检测都是核心技术。

举个例子,无人机领域,画面中的车辆、行人等目标的检测。

图解目标检测算法YOLOv1,保姆级教程。

军事上,目标的精准打击,也都有目标检测的身影。

生活上,我们也可以利用检测做很多事情,随手一拍。

图解目标检测算法YOLOv1,保姆级教程。

目标检测检测出物体,车辆、树木,可能有些人会问,这个检测出来有啥用?

那要是在此基础上,再识别出这些是什么车、什么树?

宝马,奥迪,还是比亚迪?

杨树,柳树,还是白杨树?

再比如,什么品种的猫猫狗狗,什么品牌的衣帽鞋袜。

万物皆可识别,只需随手一拍。这就是很多做视觉的公司,都在做的重点方向

这也是,目标检测+细分类的技术应用的场景之一。先检测出画面中的目标,再将检测出的目标送入更细粒度的分类网络,识别出堪比人,甚至远超人的精细水平。

目标检测,专业一点的概念就是:解决“在哪里?是什么?”的问题,即定位出这个目标的位置并且知道目标物是什么。

图解目标检测算法YOLOv1,保姆级教程。

基于深度学习的目标检测算法主要分为两类:

  • Two Stage:两阶段目标检测
  • One Stage:端到端目标检测

Two Stage

Two Stage 是2013年到2015年的主流算法,后来逐渐发展为 One Stage 端到端的目标检测算法。

Two Stage 算法是先进行区域生成,该区域称之为 region proposal(简称RP,一个有可能包含待检物体的预选框),再通过卷积神经网络进行样本分类。

任务流程:特征提取 --> 生成RP --> 分类/定位回归。

常见 Two Stage 目标检测算法有:R-CNN、SPP-Net、Fast R-CNN、Faster R-CNN 和 R-FCN 等。

One Stage

不用 RP,直接在网络中提取特征来预测物体分类和位置。

任务流程:特征提取–> 分类/定位回归。

常见的 One Stage 目标检测算法有:YOLO 系列、SSD 系列、Anchor Free 系列等。

如今,比较常用、实用的目标检测算法都是 One Stage 的。

从前,一般讲解目标检测都是从两阶段的 R-CNN 系列开始,但说实话,有些老了,一阶段的 YOLO 从 v1 都发展到 v5 了。

但并不代表 Faster R-CNN 这些就没有学的必要,目标检测的基础,还是很有必要学习一番的,并且不少公司其实还在用这些算法。

现在,一般公司面试,都爱问 YOLO 系列的算法,YOLO v4、YOLO v5 这些。

今天,先来个比较基础的,我们从 YOLO v1 谈起。

YOLO v1

YOLO 的名字还是很霸气的:You Only Look Once

YOLO 之父是 Joseph Redmon,v1 - v3 是他的作品,篇篇顶会,性能也是,开源口碑非常好。

图解目标检测算法YOLOv1,保姆级教程。

可惜,发布完 v3 就退圈了,后续的 v4 - v5 ,都是其他团队接手了。

在讲解 YOLO v1 之前,重申一下我们的任务:在一张图片中找出物体,并给出它的类别和位置。

图解目标检测算法YOLOv1,保姆级教程。

YOLO v1 的核心思想是:采用利用整张图作为网络的输入,直接在输出层回归 bounding box 的位置和 bounding box 所属的类别。

为了照顾新人,这里解释下 bounding box ,即检测框,就是目标外围带颜色的框框,一般简称 bbox。

YOLO v1 的实现,是将一幅图像分成 SxS 个网格(grid cell)。

哪个目标物体的中心落在这个网格中,则这个网格负责预测这个目标。

图解目标检测算法YOLOv1,保姆级教程。

论文中,是将图像分为 7x7 的网格,即上文中的 S=7。如上图所示,红色的点,就是负责检测狗的。

网络结构

YOLO v1目标检测一共三个步骤:

图解目标检测算法YOLOv1,保姆级教程。

  • resize图片尺寸
  • 输入网络,出结果
  • NMS

第一步,resize 很好理解,属于深度学习的常规操作,就是为了将不同尺寸的图片适配到统一的网络结构中,需要 resize 到相同的尺寸。

论文提出的这个网络结构,输入层的尺寸是 448x448,因此在将图片送入网络前,需要将图片 resize 到 448x448。

我们直接看下网络结构,精髓都在这里:

图解目标检测算法YOLOv1,保姆级教程。

这里设计得很巧妙,可以看到网络的最终输出是 7×7×30。还是以这个狗为例,7x7 很好理解,图像分为 7x7 个区域进行预测。

最终输出 tensor 的前五个数值,分别是 bbox 的 x,y,w,h,c,即 bbox 的中心坐标 x,y,bbox 的宽高 w,h,bbox 的置信度。

置信度就是算法的自信心得分,越高表示越坚信这个检测的目标没错。

而一个中心点,会检测 2 个 bbox ,这个操作可以减少漏检,因为可以适应不同形状的 bbox,进而提高bbox 的准确率。

图解目标检测算法YOLOv1,保姆级教程。

2 个 bbox 都会保留,最后通过 NMS 选择出最佳的 bbox,这个操作最后再讲。

而后面的 20 个,就是类别的概率,YOLO v1 是在 VOC 数据集上训练的,因此一共 20 个类。

图解目标检测算法YOLOv1,保姆级教程。

因此,tensor 总共加起来就是 7×7×(2×5+20),写成公式就是:SxSx(Bx5+C)。

这里的问题,可以当作回归问题来解决的。所有的输出包括坐标和宽高都定义在0到1之间。

图解目标检测算法YOLOv1,保姆级教程。

来看一下每个单元格预测的B个(x,y,w,h,confidence)的向量和C的条件概率中,每个参数的含义(假设图片宽为\(w_{i}\)高为\(h_{i}\),将图片分为\(S \times S\):

(x,y)是bbox的中心相对于单元格的offset

对于下图中蓝色框的那个单元格坐标为\((x_{c o l}=1, y_{\text {row }}=4)\),假设它预测的输出是红色框的bbox,设bbox的中心坐标为\(\left(x_{c}, y_{c}\right)\),那么最终预测出来的\((x, y)\)是经过归一化处理的,表示的是中心相对于单元格的offset,计算公式如下:

$$x=\frac{x_{c}}{w_{i}} S-x_{c o l}, y=\frac{y_{c}}{h_{i}} S-y_{\text {row }}$$

(w,h)是bbox相对于整个图片的比例

预测的 bbox 的宽高为\(\left(w_{b}, h_{b}\right)\) ,\((w, h)\)表示的是 bbox 的是相对于整张图片的占比,计算公式如下:

$$w=\frac{w_{b}}{w_{i}}, h=\frac{h_{b}}{h_{i}}$$

confidence

这个置信度是由两部分组成,一是格子内是否有目标,二是bbox的准确度。定义置信度为\(P_{r}(Object) * I O U_{\text {pred }}^{\text {truth }}\)。

这里,如果格子内有物体,则\(P_{r}(Object)=1\),此时置信度等于IoU。如果格子内没有物体,则\(P_{r}(Object)=0\),此时置信度为0。

C类的条件概率

条件概率定义为\(P_{r}\left(\right.Class_{i} \mid Object )\) ,表示该单元格存在物体且属于第 i 类的概率。

在测试的时候每个单元格预测最终输出的概率定义为:

$$
P_{r}\left(\text { Class }_{i} \mid \text { Object }\right) * P_{r}(\text { Object }) * I O U_{\text {pred }}^{\text {trrth }}=P_{r}\left(\text { Class }_{i}\right) * I O U_{\text {pred }}^{\text {truth }}
$$

NMS

经过网络处理后,将\((S \times S) \times B \times 20\) 的结果送入 NMS ,最后即可得到最终的输出框结果。

NMS,即非极大值抑制,就是将一些冗余框去掉,示意图如下:

图解目标检测算法YOLOv1,保姆级教程。

NMS 别看简单,面试常考题,比如动手实现一个 NMS 代码之类的。

这个概念千万不要懵懵懂懂,细节决定成败。省着被嘲讽:NMS都不会,做什么Detection

图解目标检测算法YOLOv1,保姆级教程。

学 NMS ,先要理解 IOU。

IOU 即Intersection over Union,也就是两个box区域的交集比上并集,下面的示意图就很好理解,用于确定两个框的位置像素距离。

图解目标检测算法YOLOv1,保姆级教程。

  • 首先计算两个box左上角点坐标的最大值和右下角坐标的最小值
  • 然后计算交集面积
  • 最后把交集面积除以对应的并集面积

就这简单,NMS 就是通过计算 IOU 来去除冗余框的,具体的实现思路如下。

以下图为例,进行说明:

图解目标检测算法YOLOv1,保姆级教程。

就像上面的图片一样,定位一个车辆,最后算法就找出了一堆的方框,我们需要判别哪些矩形框是没用的。

非极大值抑制的方法是:先假设有6个矩形框,根据分类器的类别分类概率做排序,假设从小到大属于车辆的概率 分别为A、B、C、D、E、F。

(1)从最大概率矩形框F开始,分别判断A~E与F的重叠度IOU是否大于某个设定的阈值;

(2)假设B、D与F的重叠度超过阈值,那么就扔掉B、D;并标记第一个矩形框F,是我们保留下来的。

(3)从剩下的矩形框A、C、E中,选择概率最大的E,然后判断E与A、C的重叠度,重叠度大于一定的阈值,那么就扔掉;并标记E是我们保留下来的第二个矩形框。

就这样一直重复,找到所有被保留下来的矩形框。

代码实现:

LOSS

从图片 resize 到网络结构,再到 NMS 都讲完了,最后再说下算法的重中之重,损失函数。

损失函数是一个复合损失函数,一共是 5 项相加:

图解目标检测算法YOLOv1,保姆级教程。

更详细的注释,可以看这张图:

图解目标检测算法YOLOv1,保姆级教程。

简单解释下:

所有的损失都是使用平方和误差公式,暂时先不看公式中的 \(\lambda_{\text {coord }}\) 与 \(\lambda_{\text {noobj }}\) ,输出的预测数值以及所造成的损失有:

1. 预测框的中心点\((x,y)\) 。造成的损失是图五中的第一行。其中 \(1_{i j}^{o b j}\) 为控制函数,在标签中包含物体的那些格点处,该值为 1 ;若格点不含有物体,该值为 0。也就是只对那些有真实物体所属的格点进行损失计算,若该格点不包含物体,那么预测数值不对损失函数造成影响。\((x,y)\) 数值与标签用简单的平方和误差。

2. 预测框的宽高\((w,h)\) 。造成的损失是图五的第二行。\(1_{i j}^{o b j}\) 的含义一样,也是使得只有真实物体所属的格点才会造成损失。这里对 \((w,h)\) 在损失函数中的处理分别取了根号,原因在于,如果不取根号,损失函数往往更倾向于调整尺寸比较大的预测框。例如,20 个像素点的偏差,对于 800x600 的预测框几乎没有影响,此时的IOU数值还是很大,但是对于 30x40 的预测框影响就很大。取根号是为了尽可能的消除大尺寸框与小尺寸框之间的差异。

3. 第三行与第四行,都是预测框的置信度C。当该格点不含有物体时,即 \(1_{i j}^{noo b j}\) =1, 该置信度的标签为0;若含有物体时,该置信度的标签为预测框与真实物体框的IOU数值(IOU计算公式为:两个框交集的面积除以并集的面积)。

4. 第五行为物体类别概率P,对应的类别位置,该标签数值为1,其余位置为0,与分类网络相同。

此时再来看 \(\lambda_{\text {coord }}\) 与 \(\lambda_{\text {noobj }}\) ,YOLO 面临的物体检测问题,是一个典型的类别数目不均衡的问题。其中 49 个格点,含有物体的格点往往只有 3、4 个,其余全是不含有物体的格点。此时如果不采取点措施,那么物体检测的mAP不会太高,因为模型更倾向于不含有物体的格点。 \(\lambda_{\text {coord }}\) 与 \(\lambda_{\text {noobj }}\) 的作用,就是让含有物体的格点,在损失函数中的权重更大,让模型更加“重视”含有物体的格点所造成的损失。在论文中,\(\lambda_{\text {coord }}\) 与 \(\lambda_{\text {noobj }}\) 的取值分别为 5 与 0.5 。

最后

YOLO v1 的代码实现,有很多,Pytorch 和 Tensorflow 等各种版本都有,因为毕竟属于出名算法。

Github 一搜,找个 Star 高的即可。

本期硬核,欢迎转发+再看。

我是 Jack,我们下期见~

weinxin
微信公众号
分享技术,乐享生活:微信公众号搜索「JackCui-AI」关注一个在互联网摸爬滚打的潜行者。
Jack Cui

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

目前评论:8   其中:访客  6   博主  2

    • avatar 容添下 来自天朝的朋友 火狐浏览器 Windows 7 湖北省恩施州 电信 0

      麻雀虽小五脏俱全

      • avatar 番茄味 来自天朝的朋友 谷歌浏览器 Windows 10 湖北省武汉市 中国科学院测量与地球物理研究所 0

        能不能讲述一下代码? :mrgreen: 网上没找到完整合适的v1

        • avatar 十一锟斤拷/aiq 来自天朝的朋友 谷歌浏览器 Windows 10 吉林省长春市 吉林大学 0

          请问如何实现目标检测+细分类?

          • avatar 锟斤拷十. 来自天朝的朋友 谷歌浏览器 Windows 10 河北省石家庄市 联通 0

            一名准研究生,刚看完崔哥写的yolov1的算法,真的是太棒了,保姆级的,期待崔哥出v2-5!

            • avatar Venn 来自天朝的朋友 谷歌浏览器 Windows 10 山东省威海市 移动 0

              加油加油,写的真得非常详细,期望2-5的发布。

              • avatar OK-sang 来自天朝的朋友 谷歌浏览器 Windows 10 上海市 上海医疗器械高等专科学校 0

                终于整明白了,期待yolov2-v5