yolov3和yolov4损失函数

时间:2024-04-03 13:48:10

yolov3和yolov4损失函数

yolov3损失函数

loss公式

yolov3和yolov4损失函数
其中:
1.回归loss会乘以一个(2-w*h)的比例系数,如果没有这个系数AP会下降明显,大概是因为COCO数据集小目标很多的原因。
2.置信度loss损失函数采用交叉熵,分为两部分:有物体,没有物体,其中nobody的loss还增加了权重系数,这是为了减少没有物体计算部分的贡献权重。
3.分类loss损失函数采用交叉熵,当第i个网格的第j个anchor box负责某一个真实目标时,那么这个anchor box所产生的bounding box才会去计算分类损失函数。

计算时,当第i个网格的第j个anchor box负责某一真实目标时,我们需要对这个anchor box产生的bounding box进行 求中心坐标误差,求宽高误差,求置信度误差,求分类误差。(因为这个anchor box负责了某一个目标,那么这个anchor box产生的bounding box自然很重要,计算所有的损失函数,用来更新权重;其他没有真实目标的anchor box所产生的bounding box,只求一个置信度误差。)

yolov4损失函数