The YOLO loss function is calculated in the following steps:
- First, we find the bounding boxes with the highest intersection over union (IoU) and with the correct bounding boxes.
- Then, we calculate the confidence loss, which means the probability of the object being present inside a given bounding box.
- Next, we calculate the classification loss, which indicates the present class of objects within the bounding box.
- Finally, we calculate the coordinate loss for matching the detected boxes.
In summary, the total loss function is as follows:
In the next section, we will learn about the YOLO architecture.