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

22 Nov 2023

python

이미지에서 동영상에서 선을 그리는 것은 컴퓨터 비전 작업에서 매우 유용한 기능입니다. OpenCV는 이러한 작업을 수행하기 위한 강력한 라이브러리입니다. 이번 포스트에서는 Python과 OpenCV를 사용하여 이미지에서 동영상에서 선을 긋는 방법을 알아보겠습니다.

우선, OpenCV를 설치해야 합니다. 다음 명령을 사용하여 pip를 통해 설치할 수 있습니다:

pip install opencv-python

이제 소스 코드를 작성해보겠습니다. 먼저 필요한 모듈을 가져옵니다.

import cv2

그런 다음 선을 그릴 이미지와 동영상 파일을 로드합니다.

image = cv2.imread('image.jpg')
video = cv2.VideoCapture('video.mp4')

이제 무한 루프를 사용하여 이미지의 각 프레임마다 선을 그리는 작업을 수행합니다.

while True:
    ret, frame = video.read()
    if not ret:
        break

    # 선의 시작점과 끝점을 정의합니다.
    start_point = (100, 100)
    end_point = (200, 200)

    # 선의 색상과 두께를 정의합니다.
    color = (0, 255, 0)
    thickness = 3

    # 이미지에 선을 그립니다.
    image_with_line = cv2.line(frame, start_point, end_point, color, thickness)

    # 선이 그려진 이미지를 화면에 표시합니다.
    cv2.imshow("Line Drawing", image_with_line)

    # 'q' 키를 누르면 종료합니다.
    if cv2.waitKey(1) == ord('q'):
        break

video.release()
cv2.destroyAllWindows()

위의 코드에서는 video.read()를 사용하여 동영상의 각 프레임을 읽고, cv2.line()을 사용하여 이미지에 선을 그립니다. 그리고 cv2.imshow()를 사용하여 선이 그려진 이미지를 화면에 표시합니다. 마지막으로 cv2.waitKey()를 사용하여 윈도우를 업데이트하고, ‘q’ 키를 누르면 프로그램을 종료합니다.

이제 이 코드를 실행해보세요. 지정된 이미지에서 동영상의 각 프레임마다 선이 그려진 결과를 확인할 수 있어야 합니다.

본 포스트에서는 Python과 OpenCV를 사용하여 이미지에서 동영상에서 선을 그리는 방법을 알아보았습니다. OpenCV는 더 많은 컴퓨터 비전 작업을 수행할 수 있도록 다양한 기능을 제공하므로, 관심 있는 분들은 더 공부해보시기 바랍니다.

참고:

OpenCV 문서: https://docs.opencv.org/4.5.1/
OpenCV-Python 문서: https://docs.opencv.org/4.5.1/d6/d00/tutorial_py_root.html