[python] 파이썬을 사용하여 동영상에서 물체 추적하기

14 Dec 2023

python

동영상에서 물체를 추적하는 것은 컴퓨터 비전 및 영상처리 분야에서 흔한 작업입니다. 여기에는 OpenCV와 같은 파이썬 라이브러리를 활용할 수 있습니다.

동영상 파일에서 물체 추적하기

우선 OpenCV 라이브러리를 설치합니다. 다음은 동영상 파일을 불러온 후 물체를 추적하는 간단한 예제 코드입니다.

import cv2

video_file = 'video.mp4'
tracker = cv2.TrackerCSRT_create()

video = cv2.VideoCapture(video_file)
ok, frame = video.read()
bbox = cv2.selectROI(frame, False)
tracker.init(frame, bbox)

while True:
    ok, frame = video.read()
    if not ok:
        break
    
    # 물체 추적
    ok, bbox = tracker.update(frame)
    
    if ok:
        # 추적 결과 시각화
        p1 = (int(bbox[0]), int(bbox[1]))
        p2 = (int(bbox[0] + bbox[2]), int(bbox[1] + bbox[3]))
        cv2.rectangle(frame, p1, p2, (0,0,255), 2, 1)
    else:
        # 추적 실패 시 예외 처리
        pass
        
    cv2.imshow('Tracking', frame)
    
    if cv2.waitKey(1) & 0xFF == 27:  # ESC 키로 종료
        break

video.release()
cv2.destroyAllWindows()

위 코드는 OpenCV의 TrackerCSRT를 사용하여 물체를 추적하는 간단한 예제입니다. 이를 실행하면 동영상 파일에서 물체를 추적하는 화면이 표시됩니다.

추가로 읽을 거리

OpenCV 공식 문서: OpenCV 공식 문서
파이썬 영상 처리 및 컴퓨터 비전을 위한 OpenCV 3 프로그래밍 입문: https://opencv-python.readthedocs.io/en/latest/
파이썬 공식 문서: 파이썬 공식 문서