其他分享
首页 > 其他分享> > Yolo V1原理及应用

Yolo V1原理及应用

作者:互联网

 

YOLO采用卷积网络来提取特征,然后使用全连接层来得到预测值。网络结构参考GooLeNet模型,包含24个卷积层和2个全连接层,如图4所示。对于卷积层,主要使用1x1卷积来做channel reduction,然后紧跟3x3卷积。对于卷积层和全连接层,采用Leaky ReLU激活函数:。但是最后一层却采用线性激活函数

 

 可以看到网络的最后输出为7*7*30大小的张量。这和前面的讨论是一致的,这个张量所代表的具体含义如图9所示。对于每一个单元格,前20个元素是类别概率值,然后2个元素是边界框置信度,两者相乘可以得到类别置信度,最后8个元素是边界框的[公式]。大家可能会感到奇怪,对于边界框为什么把置信度[公式][公式]都分开排列,而不是按照[公式]这样排列,其实纯粹是为了计算方便,因为实际上这30个元素都是对应一个单元格,其排列是可以任意的。

 

综上所述,YOLO的最终误差为下:

Loss = λcoord * 坐标预测误差 + (含object的box confidence预测误差 + λnoobj * 不含object的box confidence预测误差) + 类别预测误差

 

 

 

 

标签:单元格,置信度,边界,预测值,Yolo,V1,类别,原理,预测
来源: https://blog.csdn.net/SerendipityZYS/article/details/120617272