Search code examples
object-detectionyolodarknetyolov4

Which one is the correct YOLOv4 total loss function formula?


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


Solution

  • 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.