根据pytorch的代码,Yolov3的损失函数主要包括3个部分:
(1)对于正例,计算bbox与GT之间的位置和大小差异,采用MSE损失函数;
(2)对于正例,计算80个类别维度与target的one-hot向量间的交叉熵损失;
(3)对于正例和负例,分别计算置信度与真实之间的交叉熵,并按照一定的比例系数加权求和。
根据pytorch的代码,Yolov3的损失函数主要包括3个部分:
(1)对于正例,计算bbox与GT之间的位置和大小差异,采用MSE损失函数;
(2)对于正例,计算80个类别维度与target的one-hot向量间的交叉熵损失;
(3)对于正例和负例,分别计算置信度与真实之间的交叉熵,并按照一定的比例系数加权求和。