深度学习·神经网络·计算机视觉人工智能时代(AI)深度学习目标检测

【算法】YOLOv1,v2,v3的区别

2018-09-05  本文已影响173人  yuanCruise
1.损失函数(各分量之间的系数)

损失函数:


image

三个版本损失函数系数详情:

versions coord obj class noobj
yolov1 1 0.5 1 5
yolov2 5 1 1 1
yolov3 1 1 1 gt相关
2.网络输出形式

按照损失行数的行数来说明loss需要的标签计算。
一:localization,坐标误差。(标签不用计算,直接从输入中获取)
二:含有object的box置信度误差。(标签需要计算)
三:不含有object的box置信度误差。(标签需要计算)
四:含有object的cell的那个“对应”box计算得到的类条件概率误差。(标签需要计算)


如何计算二,三两个标签,通过以下几个问题来回答~
问:什么是当前cell的所有box的confidence呢?
答:每一个box都会有一个confidence。
每一个box 的confidence计算公式:
Pr(Object)*IOUtruth/pred。
所以根据公式可知,如果当前cell中没有object,那么Pr(Object)=0,所以confidence=0,;如果当前cell中有object,那么Pr(Object)=0,所以confidence=IOUtruth/pred。
最后的confidece_label的标签是0和1,为0的有两种情况,第一就是上面所说的Pr(Object)=0,第二是IOUtruth/pred的阈值小于规定的值。而为1的就是IOUtruth/pred大于阈值。


如何计算四个标签,通过以下几个问题来回答~
问:那么为什么还要说是当前 cell 对应的 box 呢?
答:因为每一个cell在初始化的时候生成的初始化box是大于1个的,而最终参与类条件概率计算的时候只会用到一个 box,所以要说明对应的box。


问:那么怎么样从初始化的多个box中选出这个"对应"的box呢?
答:由于这多个box都是属于当前cell的,而每个cell都会负责小于等于一个目标(ground truth)。这多个初始化box与当前cell负责的ground truth计算IOU,那么拥有max_IOU的那个box就是而这个“对应”的box。因此这一个box就能够代表当前cell了。


问:那要是当前cell负责目标的个数为0,那怎么确定这个“对应”的box呢?
答:我的理解是当,当前cell负责的目标个数为0的时候,这个cell所有的box的confidence都=0,而且类条件概率也都是0。所以这个时候不需要知道这个“对应”的box是谁。


问:怎么确定当前cell是负责目标的,还是不负责目标的呢?
答:如果一个目标的ground truth的中心点坐标在这个cell中,那么这个cell就是包含这个目标,也就是说这个目标的预测就由该cell负责。


问:最后我得到了这个IOU最大的“对应”的box之后,怎么通过这个box计算我所需要的第四个标签呢?
答:这个标签其实就是判断一下当前cell负责的是哪一个object,那么对应的那个类概率就是1,其余都是0。所以。。。。。其实这个标签是直接可以通过类别和cell的位置来得到的,并不需要通过这个对应的box来得到。相当于每一个cell都包含一个类标签。当然和这个标签做loss的时候是要确定这个对应的box的。所以上述的解释对于理解还是有用的。

3.检测框初始化策略(anchor)

v1:随机暴力初始化
v2,v3:设置初始化anchor,在anchor基础上回归(而且这个anchor根据kmean算法进行聚类得到,这样使得初始化的anchor的比例和大小能够和当前的数据特征匹配。)


YOLOv1是利用全连接层直接预测bounding box的坐标,而YOLOv2借鉴了Faster R-CNN的思想,引入anchor。

YOLOv1将输入图像分成77的网格,如果每个cell预测两个box,因此一共只有98个box,但是在YOLOv2通过引入anchor boxes,预测的box数量超过了1千(以输出feature map大小为1313为例,每个cell有9个anchor box的话,一共就是13139=1521个)。


还有一个v1和v2,v3的差别,因为v2,v3有anchor,所以回归anchor坐标有自己的特点,具体查看这:anchor部分

4.yolov3做的一些改进

见csdn博客。


本人csdn博客地址:https://blog.csdn.net/qiu931110/
本人微信公众号:yuanCruise

上一篇下一篇

猜你喜欢

热点阅读