이 튜토리얼에서는 OpenCV의 rectangle() 함수를 사용하여 이미지 위에 사각형을 그리는 방법을 알아봅니다. 이 함수는 시작 좌표(start point), 종료 좌표(end point), 색상(color), 선 두께(thickness), 그리고 대상 이미지 자체를 매개변수로 받아 지정된 영역에 사각형을 그려줍니다.
원본 이미지
알고리즘
1단계: cv2 모듈을 임포트한다. 2단계: imread() 함수로 이미지를 불러온다. 3단계: 사각형의 시작 좌표를 정의한다. 4단계: 사각형의 종료 좌표를 정의한다. 5단계: 색상(BGR 값)과 선 두께를 정의한다. 6단계: cv2.rectangle() 함수를 호출하여 사각형을 그린다. 7단계: 결과 이미지를 화면에 표시한다.
주요 매개변수 설명
- start_point: 사각형의 왼쪽 위 꼭짓점 좌표 (x, y)
- end_point: 사각형의 오른쪽 아래 꼭짓점 좌표 (x, y)
- color: BGR 형식의 색상 값, 예: (0, 0, 255)는 빨간색
- thickness: 사각형 테두리의 두께(픽셀 단위)
예제 코드
import cv2
image = cv2.imread('testimage.jpg')
height, width, channels = image.shape
start_point = (0, 0)
end_point = (width, height)
color = (0, 0, 255)
thickness = 5
image = cv2.rectangle(image, start_point, end_point, color, thickness)
cv2.imshow('Rectangle', image)
cv2.waitKey(0)위 코드에서는 image.shape를 통해 이미지의 높이와 너비를 구한 뒤, 시작점을 (0, 0)으로, 종료점을 (너비, 높이)로 설정하여 이미지 전체를 감싸는 사각형을 그립니다. 색상은 BGR 순서로 지정되며, (0, 0, 255)는 빨간색을 의미합니다.
실행 결과
코드를 실행하면 원본 이미지 위에 빨간색 테두리의 사각형이 그려진 것을 확인할 수 있습니다. 참고로 cv2.waitKey(0)를 추가하면 창이 즉시 닫히지 않고 키 입력이 있을 때까지 결과 화면이 유지됩니다.