[python] OpenCV를 사용하여 이미지에서 선 긋기

22 Nov 2023

python

이미지 처리 및 컴퓨터 비전 작업에 가장 널리 사용되는 라이브러리 중 하나인 OpenCV를 사용하여 이미지에서 선을 그어보겠습니다.

먼저, OpenCV를 설치해야 합니다. 아래의 명령을 사용하여 pip를 통해 설치할 수 있습니다.

pip install opencv-python

설치가 완료되면 다음과 같이 코드를 작성합니다.

import cv2
import numpy as np

# 이미지 읽기
image = cv2.imread('image.jpg')

# 선의 시작점과 끝점 지정
start_point = (50, 50)
end_point = (200, 200)

# 선의 색상 및 두께 설정
color = (0, 255, 0)  # (B, G, R)
thickness = 2

# 이미지에 선 그리기
image = cv2.line(image, start_point, end_point, color, thickness)

# 결과 이미지 보여주기
cv2.imshow('Image with Line', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

위의 코드에서는 cv2.imread 함수를 사용하여 이미지를 읽은 후, cv2.line 함수를 사용하여 이미지에 선을 그립니다. 선의 시작점과 끝점은 튜플로 지정하고, 선의 색상과 두께를 설정할 수 있습니다.

마지막으로 cv2.imshow 함수를 사용하여 결과 이미지를 보여주고, cv2.waitKey 함수를 사용하여 키 입력을 대기하며, cv2.destroyAllWindows 함수를 사용하여 모든 창을 닫습니다.