太原网站建设设计,北碚网站建设哪家好,做设计的需要网站下载素材吗,最大的网站建设公司排名基于深度学习的运动目标检测#xff08;一#xff09; 1.YOLO算法检测流程2.YOLO算法网络架构3.网络训练模型3.1 训练策略3.2 代价函数的设定 2012年#xff0c;随着深度学习技术的不断突破#xff0c;开始兴起基于深度学习的目标检测算法的研究浪潮。
2014年#xff0c;… 基于深度学习的运动目标检测一 1.YOLO算法检测流程2.YOLO算法网络架构3.网络训练模型3.1 训练策略3.2 代价函数的设定 2012年随着深度学习技术的不断突破开始兴起基于深度学习的目标检测算法的研究浪潮。
2014年Girshick等人首次采用深度神经网络实现目标检测设计出R-CNN网络结构实验结果表明在检测任务中性能比DPM算法优越。同时何恺明等人针对卷积神经网络(Convolutional Neural NetworkCNN)计算复杂度高的问题引入空间金字塔池化层设计出基于SPP-Net的目标检测网络不但提高了目标检测速度而且支持任意尺寸大小的图像输入。
2015年Girshick在R-CNN目标检测网络的基础上针对候选框特征重复提取进行优化提出了Fast R-CNN(Fast Regions with CNNFast R-CNN)网络结构实现了端对端式的训练并且所有网络层的参数在不断更新。任少卿和何恺明等人采用区域建议网络(Region Proposal NetworkRPN)和Fast R-CNN网络结合的方式设计出新的目标检测网络该网络使得目标检测精度和速度得到很大的提升。
传统的目标检测算法的基本思路一般都是先对图像进行预处理然后使用滑动窗口策略在整个图像均匀间隔的区域上提取特征最后利用机器学习中的分类器判断是否存在目标比如之前的DMP系统不但检测流程复杂而且检测过程中卷积计算量较大无法满足现实生活中检测的实时性。对于该算法的不足Joseph Redmon等人提出了YOLO算法主要利用的是将目标检测任务的问题转换为回归问题其设计的网络只要进行一次计算就能直接得到完整图像中目标的边界框和类别概率。同时该网络结构单一实现端对端式的训练。因此YOLO检测算法的检测速度可满足实际的检测需求。下面我们从检测流程、网络架构、网络训练模型3个角度来分析YOLO算法。
基于YOLO算法的目标检测流程大致可分为3个步骤
(1)把待检测的图像调整大小为448×448。
(2)将图像放到CNN输出待检测目标边界框的坐标信息和类别概率。
(3)使用非极大抑制算法去除冗余的标注框筛选出最终的目标检测结果。 1.YOLO算法检测流程
YOLO算法的整体检测思路为首先将待检测图像划分为S×S个网格然后对每个网格都预测B个边界框(bounding boxes)和这些边界框所对应的自信得分(confidence scores)。YOLO算法具体的检测示意图如图所示。 此时若有目标的中心点落在某个格子单元中则该格子将负责检测这个目标而confidence score负责检测该网络模型的单元格中是否有目标以及对预测位置的精确度。当单元格中没有目标时confidence score为0如果存在目标则为预测的boxes与真实的boxes之间的IoU值。
每个边界框都包含了5个预测值xywhconfidence。其中坐标(x,y)代表边界框的中心坐标与网格单元对齐相当于当前网格单元的偏移值使得范围变成[0,1]坐标(w,h)代表预测的边框相对于整个图像的高度和宽度的比例。每个网格还要预测一个类别信息记为C类则C个类别条件概率值为Pr( | )。在测试阶段每个边界框的具体类别的自信得分计算公式如下 上式得到的结果中既包含了边界框中预测类别的概率信息也反映了边界框中是否含有目标和边界框位置的精确度。
2.YOLO算法网络架构
通过上述分析YOLO的检测流程可以看出YOLO算法将格子数S设置为7×7的大小网络方面采用GooleNet的思想其中包含了24个卷积层和2个全连接层如图所示。卷积层主要用来提取特征全连接层主要用来预测类别概率和坐标与GooleNet的不同之处在于YOLO检测网络没有采用Inception结构而采用1×1和3×3的卷积层代替了Inception。从上述网络检测流程的分析可以看出每个单元需要预测(B×5C)值假设将输入图像划分为S×S个网格单元那么最终的预测值为S×S×(B×5C)大小的张量对于PASCAL VOC数据最终的预测结果为7×7×30大小的张量。 可以看出网络的最终输出为7×7×30大小的张量这和前面的分析一致。该张量所代表的具体含义如图所示。对于每一个单元格最后20个元素代表类别概率值其中前面10个中有两个是边界框置信度两者的乘积就是类别置信度剩下8个元素代表边界框的(x,y,w,h)。 3.网络训练模型
3.1 训练策略
在训练之前先在ImageNet上进行预训练其预训练的分类模型采用图15-12中前20个卷积层然后添加一个average-pool层和全连接层。预训练之后在预训练得到的20层卷积层之上加上随机初始化的4个卷积层和2个全连接层。由于检测任务一般需要更高清的图片因此将网络的输入从224×224增加到448×448。整个网络的流程如图所示。 YOLO算法开始并没有像Goole Net那样直接构建24层的卷积网络而是构建20层的卷积网络然后在ImageNet数据集上预训练了前面的20层卷积层。Ren等人采用将卷积层和全连接层添加到预训练网络中的策略进一步提高了网络的性能。因此YOLO采用基于该原理在初始的20层卷积网络基础上随机初始化后面的4层卷积层和2层全连接层。由于在检测中需要更详细的视觉信息因此YOLO将网络模型的输入图像大小由224×224调整为448×448。网络最终预测输出结果的边界框的信息是(x,y,w,h)但是坐标(x,y)用对应网格的偏移量表示然后归一化到(0,1)之间坐标(w,h)用图像width和height归一化到(0,1)之间。由于在训练的过程中一般的激活函数Sigmoid会出现梯度消失、梯度爆炸的现象因此YOLO采用LeakyReLU线性激活函数作为激活函数其函数表达式如下 3.2 代价函数的设定
YOLO算法的最终输出结果包含边界框的坐标、置信度得分、类别概率值3个部分代价函数的设计目标是让这三方面达到很好的平衡。刚开始YOLO简单地采用平方和误差作为代价函数通过梯度下降法很容易达到最优化但是这种代价函数的设计最终导致的结果是模型的拟合能力很差主要原因在于代价函数的设定不合适同时造成的不足之处有以下两个方面
(1)8维的坐标误差和20维的类别概率误差采用平方差误差时两者视为同等重要这显然是不合理的。
(2)在一幅图像中如果一个网络中没有目标就将这些网络中的格子单元的置信度设置为0。相比于较少的有目标的网络这种方法是不可取的会导致网络不稳定甚至不收敛。基于上述的不足YOLO进一步改进代价函数更重视8维坐标预测的误差损失对其赋予更大的权值同时减小不包含目标的单元格的置信度预测的权值。对于包含目标的单元格的置信度和类别概率预测的误差损失保持不变同时将其权值设定为1。平方和误差同时对大边界框和小边界框中的误差采用相同的权值改进后小边界框中的微小偏差显得尤为重要。因此网络的代价函数将原来的w、h分别用[插图]和[插图]代替这样做的好处在于进一步降低了敏感度之间的差异。最终代价函数设计如下 其中第一项是边界框中心坐标的误差项[插图]用于判断第i个网格中第j个边界框是否检测该目标λcoord表示边界框坐标误差损失的权值[插图]用于判断目标是否出现在网格中λnoobj表示不包含目标边界框的置信度预测的误差权值。在上述代价函数中只有当某个网络中有目标时才对预测误差进行惩罚即增加权重系数。
总之YOLO算法的缺点有两个方面一方面难以检测小目标导致对视频中运动目标的检测出现漏检的现象另一方面对目标的定位不准检测到的目标与实际目标之间存在偏差。
