Hi @daminglu
Yes, you are correct! The visualization still shows real objects because each 19x19 grid cell shows the class with the highest predicted probability, mapped back over the resized input image (yes, it can be rendered using yolo_head, but it’s not part of YOLO’s main prediction logic).
Hope it helps!