OpenCV는 컴퓨터 비전 및 이미지 처리를 위한 강력한 라이브러리입니다. 이 라이브러리를 사용하면 이미지에 다양한 도형을 그릴 수 있습니다. 이번 글에서는 Python과 OpenCV를 사용하여 이미지에 도형을 그리는 방법을 알아보겠습니다.

1. OpenCV 설치하기

먼저, OpenCV를 설치해야 합니다. 아래 명령어를 사용하여 OpenCV를 설치하세요.

pip install opencv-python

2. 이미지 로드하기

먼저, 이미지를 로드해야 합니다. 다음은 OpenCV를 사용하여 이미지를 로드하는 코드입니다.

import cv2

image = cv2.imread('image.jpg')

로드한 이미지는 image 변수에 저장됩니다. 원하는 이미지 파일의 경로를 imread 함수에 전달하면 됩니다.

3. 도형 그리기

이제 이미지에 도형을 그릴 차례입니다. 다음은 몇 가지 도형을 그리는 예제 코드입니다.

선 그리기

cv2.line(image, (x1, y1), (x2, y2), (255, 0, 0), thickness)

위 코드에서 (x1, y1)은 선의 시작점 좌표이고, (x2, y2)는 선의 끝점 좌표입니다. (255, 0, 0)은 선의 색상(RGB 형식)이며, thickness는 선의 두께입니다.

사각형 그리기

cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), thickness)

위 코드에서 (x1, y1)은 사각형의 왼쪽 위 모서리 좌표이고, (x2, y2)은 사각형의 오른쪽 아래 모서리 좌표입니다. (0, 255, 0)은 사각형의 색상(RGB 형식)이며, thickness는 사각형의 선 두께입니다.

원 그리기

cv2.circle(image, (center_x, center_y), radius, (0, 0, 255), thickness)

위 코드에서 (center_x, center_y)는 원의 중심 좌표이고, radius는 원의 반지름입니다. (0, 0, 255)은 원의 색상(RGB 형식)이며, thickness는 원의 선 두께입니다.

4. 이미지 저장하기

도형을 그린 후에는 이미지를 저장할 수 있습니다. 다음은 이미지를 저장하는 코드입니다.

cv2.imwrite('output.jpg', image)

위 코드에서 'output.jpg'는 저장할 이미지 파일의 경로입니다.

전체 코드

import cv2

# 이미지 로드
image = cv2.imread('image.jpg')

# 선 그리기
cv2.line(image, (x1, y1), (x2, y2), (255, 0, 0), thickness)

# 사각형 그리기
cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), thickness)

# 원 그리기
cv2.circle(image, (center_x, center_y), radius, (0, 0, 255), thickness)

# 이미지 저장
cv2.imwrite('output.jpg', image)

이상으로 OpenCV를 사용하여 이미지에 도형을 그리는 방법을 알아보았습니다. OpenCV에는 다양한 도형 그리기 함수 및 옵션이 있으니 필요에 따라 적절한 함수를 사용하시기 바랍니다. OpenCV 공식 문서를 참조하면 더 자세한 정보를 얻을 수 있습니다.