YOLO grid cell Bounding Box


Object Detection에서 Bounding Box를 다룰 때 크게 두 가지 coordinate로 Box를 다룬다.

특히 IoU(loss)를 다룰 때는 corner coordinate를 사용한다.

anchor나 MSELoss를 다룰 때는 center coordinates 를 사용한다.

Bounding Box coordinate

bounding box coordinates

bounding box coordinates

파란 색 : bounding box를 좌상점과 우하점으로 좌표를 나타내는 corner coordinates 방식 빨간 색 : bounding box를 중심점과 width, height로 좌표를 나타내는 center coordinates 방식


참고 링크

center_to_corner, corner_to_center Pytorch 코드

[Object Detection] bounding box coordinates 설명과 pytorch 구현