[python] 파이썬을 이용하여 동영상에 자동 트랙킹 추가하기

08 Dec 2023

python

동영상에 객체를 자동으로 추적하고 싶다면 파이썬의 여러 라이브러리를 이용할 수 있습니다. 이 글에서는 OpenCV와 Dlib 라이브러리를 사용하여 파이썬으로 동영상에서 객체를 추적하는 방법을 살펴보겠습니다.

1. OpenCV와 Dlib 라이브러리 설치하기

먼저, OpenCV와 Dlib 라이브러리를 설치해야 합니다. 다음 명령어를 사용하여 필요한 라이브러리를 설치할 수 있습니다.

pip install opencv-python dlib

2. 동영상에서 객체 추적하기

다음은 파이썬을 사용하여 동영상에서 객체를 추적하는 간단한 예제 코드입니다.

import cv2
import dlib

# 동영상 읽기
cap = cv2.VideoCapture('input_video.mp4')

# 객체 추적기 생성
tracker = dlib.correlation_tracker()

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

    # 추적할 객체 선택 (사각형 좌표 설정)
    rect = dlib.rectangle(left, top, right, bottom)
    
    # 추적기에 현재 프레임과 bounding box 전달
    tracker.start_track(frame, rect)

    # 추적된 객체에 사각형 그리기
    tracked_position = tracker.get_position()
    left = int(tracked_position.left())
    top = int(tracked_position.top())
    right = int(tracked_position.right())
    bottom = int(tracked_position.bottom())
    cv2.rectangle(frame, (left, top), (right, bottom), (0, 255, 0), 2)

    # 화면에 표시
    cv2.imshow('Frame', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

3. 추가 학습 및 자원

OpenCV 공식 문서: https://docs.opencv.org/
Dlib 공식 문서: http://dlib.net/

위 예제를 참고하여 파이썬을 사용하여 동영상에서 객체를 추적하는 방법을 익혔다면, 추가로 OpenCV와 Dlib 문서를 참조하여 더 많은 기능을 학습할 수 있습니다.