YOLO annotation

YOLO에서 annotation은 다음과 같이 한다. 가장 먼저 명심해두어야 하는 것은 좌상단을 (0,0), 우하단이 (1,1)인 사각형 형태이며, 비율로 해야한다는 것이다.

  • 좌상단을 (0,0), 우하단이 (1,1)인 사각형 형태이다.
  • 입력되는 값은 비율에 해당하는 값이다.
  • txt 포맷(TSV)로 입력할 수 있다.
  • 첫 번째는 클래스
  • 두 번째는 형태 중심의 X
  • 세 번째는 형태 중심의 Y
  • 네 번째는 가로 폭
  • 다섯 번째는 세로 폭

OpenCV에서 boundingRect 를 이용할 경우 출력 값은 Contour의 좌상단 X좌표, Y좌표, 폭, 높이이다. 보통 x y, w, h로 결과를 받는다.

x, y, w, h = cv2.boundingRect(contour)

간단한 식으로 결과를 변환하여 입력해주면 된다.