As you can see there is output_1_loss and output_2_loss. This is also computed for validation and any extra metric I add. I get metric_name_output_1 and metric_name_output_2.
I would assume there are two output layers, but it seems like in the loss the function that all the information in yolov4 is already contained in one:
Hey, It seems that the loss function is computed for output_1 and output_2.
This can be seen by setting verbose to 1 and seeing the keras traning output:
As you can see there is
output_1_loss
andoutput_2_loss
. This is also computed for validation and any extra metric I add. I getmetric_name_output_1
andmetric_name_output_2
.I would assume there are two output layers, but it seems like in the loss the function that all the information in yolov4 is already contained in one:
Those are the 3 outputs expected from YOLOv4
source: https://github.com/hhk7734/tensorflow-yolov4/blob/master/py_src/yolov4/tf/train.py#L71