Do you understand why the Outputs of my YOLOV4 model has this shape? [duplicate]

Question

I have a question. I trained a YOLOV4 model for face detection and when i tried to look at the output on Neutron i found that the Bounding Box shape is [1,1,1] and Feature has the shape of [1,1,4] is there something wrong that i did ?

Thank you![Neutron] https://i.stack.imgur.com/jnj1h.png

Answer 1

This is already answered here. But basically it's like this:

(1,1,4) is the result for the bounding boxes. The first number 1 is according to your image fed into the model. As you have only one object then the output is 1 for the second number. Furthermore, YOLO configuration for bounding boxes are (x_center,y_center,width,height), which are the 4 positions.

(1,1,1) will be the same, but now 1 is for the label of the class you choose.