1

I couldn't find the total loss function in the main YOLOv4 paper. However, I found two differing formulas from two different papers (difference is highlighted/marked in the images below). Which formula is the correct default total loss formula for YOLOv4?

FORMULA 1 - Source FORMULA 1

FORMULA 2 - Source enter image description here

Addie Ira B. Parico
  • 607
  • 4
  • 25
  • I don't have an answer. But in grepping the Darknet source code, there seems to be a lot of mention of loss in yolo_layer.c. Would the source code tell you which is correct? – Stéphane Apr 02 '21 at 08:14

1 Answers1

0

Formula 1 is the more correct one because that term of the formula is the confidence loss when there is an object detected. And the confidence loss is based on the cross entropy error.

Addie Ira B. Parico
  • 607
  • 4
  • 25