YOLOv1: You Only Look Once: Unified, Real-Time Object Detection

HQL_HQL

已于 2022-07-24 15:16:14 修改

阅读量754

点赞数 1

文章标签： cnn 目标检测深度学习人工智能

于 2022-07-05 13:32:22 首次发布

本文链接：https://blog.csdn.net/m0_55308155/article/details/125606495

版权

1. 论文简介

Look

论文名：You Only Look Once: Unified, Real-Time Object Detection

论文地址：YOLOv1

论文作者：Joseph Redmon, Santosh Divvala, Ross Girshick , Ali Farhadi

论文时间：2016年CVPR

2. 网络结构

YOLOv1的网络结构借鉴了GooLeNet，共包含24个卷积层，2个全连接层，其中，3x3的卷积后接1x1 reduction layer来取代GooLeNet的inception module。

输入：448 x 448 x 3

前向传播：24个Conv Layer + 2个FC，3x3的卷积后接1x1的卷积来做通道上的降维，最后一个FC层包含1470个神经元，得到1470x1的输出，经过reshape后得到7 x 7 x 30的tensor。卷积层和全连接层采用Leaky ReLU激活函数，最后一层采用线性激活函数。

输出：最后一层FC的输出经过reshape后得到7 x 7 x 30的tensor，也就是，每个grid cell都有一个30维的输出，代表包含2个bounding box的信息和20个类别概率，下文马上介绍。

3. 7 x 7 x 30的解释

（1）YOLOv1首先会将图片划分为SxS个grid cell（网格），如果某个object的中心落在这个网格中，那么这个网格就负责预测这个object。论文中取S=7。例如，下图中，狗的中心点所在的grid cell就负责预测这只狗。

（2）每个网格预测B个bounding box。论文中取B=2。

具体的：

每个bounding box的预测信息包含位置和confidence值m，也就是，每个bounding box的预测信息包含5个值：x，y，w，h，confidence。

confidence值是YOLO系列网络特有的一个值，它衡量了模型以多大的置信度认为这个box包含了目标物体以及这个box的准确率有多高。

$confidence = Pr(Object)*IOU_{pred}^{truth}$

当bounding box中存在目标时，Pr(Object)=1，此时confidence=IOU（交并比）；当bounding box中不存在目标时，Pr(Object)=0，此时confidence=0。

x，y是bounding box的中心点坐标，它是相对于grid cell而言的值（所以是0-1之间的值），w,h是bounding box相对于整个图像的宽和高的相对值（所以也是0-1之间的值），即w=bounding box的宽/图片的宽，h=bounding box的高/图片的高。（YOLOv1中没有anchor的概念，是直接预测这几个值的）

每个grid cell还会预测C个条件类别概率， $Pr(Class_{i}|Object)$ ，它表示在bounding box包含了目标的情况下属于C个类别的概率。这里需要注意，虽然每个grid cell会预测B个bounding box，但是每个grid cell只预测一组类别概率。

在测试阶段，会将条件概率和confidence值相乘。条件概率即：每个bounding box分别属于每个类的概率值，例如Pascal VOC数据集有20个类，那么这里的概率值就有20个，公式为：

$Pr(Class_{i}|Object)*Pr(Object)*IOU_{pred}^{truth}=Pr(Class_{i})*IOU_{pred}^{truth}$

也就是测试阶段输出的概率分数即包含了这个bounding box内包含某个目标的概率，也包含了这个bounding box与真实目标的重合程度。

于是！！！最终的输出为7x7x30。（Pascal VOC数据集一共有20类）

输入图片被分成7x7个网格，每个网格预测输出2个bounding box，每个bounding box包含5个值：4个坐标+1个置信度，另外每个网格预测输出20个类别（针对Pascal VOC数据集），所以最终预测输出的维度为：7 x 7 x (4 + 1 + 4 + 1 + 20) = 7 x 7 x 30，如下图所示。

4. 损失函数

包含了：bounding box损失+confidence损失+classes损失。

主要采用：sum-square error，因为优化起来比较方便。

损失函数中，bounding box损失的系数 $\lambda _{coord}=5$ ，不包含目标的confidence系数 $\lambda _{noobj}=0.5$ ，这是因为：在一张图片中，很多grid cell并不包含目标，这会导致confidence值为0，影响包含了目标的这些grid cell的梯度的作用，也会导致模型不稳定，过早发散。

bounding box损失：

1）并不是所有的bounding box都参与损失的计算。满足：第i个grid cell中包含目标，同时，该grid cell中第j个bounding box和ground truth有最大的IoU值，那么这个bounding box才参与loss的计算，其他的不满足条件的bounding box不参与损失的计算。

2）位置损失部分，宽和高取开根号处理。作者在论文中提出，损失是应该要反映出小偏移量对于大目标带来的影响是小于小目标的。具体分析如下：假设有两个不同大小的目标，在两个bounding box有相同的偏移量时，小目标相对于大目标来说，这个bounding box的预测效果明显更差，如左图所示。而在采用开根号之后，相同的偏移量下，小目标带来的损失明显是大于大目标的，对比右边两图可知。