파이썬과 OpenCV를 사용하여 동영상에서 물체 속도 측정하기

17 Oct 2023

opencv

물체의 속도를 측정하는 것은 컴퓨터 비전 분야에서 중요한 주제 중 하나입니다. 이번 포스트에서는 파이썬과 OpenCV를 사용하여 동영상에서 물체의 속도를 측정하는 방법을 알아보겠습니다.

1. 필요한 라이브러리 설치

이 작업을 위해 우선 파이썬과 OpenCV를 설치해야 합니다. 아래 명령을 사용하여 필요한 라이브러리를 설치합니다.

pip install opencv-python

2. 동영상 불러오기

이제 속도를 측정할 동영상을 불러와야 합니다. 이를 위해 OpenCV의 VideoCapture 클래스를 사용합니다. 다음 코드를 사용하여 동영상을 불러올 수 있습니다.

import cv2

video = cv2.VideoCapture('video.mp4')

3. 물체 추적하기

물체의 속도를 측정하기 위해서는 물체를 추적해야 합니다. OpenCV의 Tracker 클래스를 사용하여 물체를 추적할 수 있습니다. 다음 코드는 사각형을 사용하여 물체를 추적하는 방법을 보여줍니다.

tracker = cv2.TrackerCSRT_create()
success, frame = video.read()
bbox = cv2.selectROI("Tracking", frame, False)
tracker.init(frame, bbox)

4. 프레임 처리하기

이제 추적된 물체를 사용하여 속도를 측정할 수 있습니다. 동영상의 각 프레임을 읽고 물체를 추적한 다음, 속도를 계산합니다. 다음 코드는 이를 구현한 예시입니다.

while True:
    success, frame = video.read()
    if not success:
        break
    
    success, bbox = tracker.update(frame)
    
    if success:
        # 물체의 위치 변경 및 속도 계산
        x, y, w, h = [int(v) for v in bbox]
        center_x = x + w // 2
        center_y = y + h // 2
        velocity = calculate_velocity(center_x, center_y)  # 물체의 속도 계산 함수 호출
        
        # 화면에 속도 출력
        cv2.putText(frame, f"Velocity: {velocity} px/frame", (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)
    
    cv2.imshow("Tracking", frame)
    if cv2.waitKey(1) == ord('q'):
        break

video.release()
cv2.destroyAllWindows()

5. 결론

이제 파이썬과 OpenCV를 사용하여 동영상에서 물체의 속도를 측정하는 방법을 알아보았습니다. 이를 활용하여 다양한 물체의 속도를 측정하고 분석할 수 있습니다.

#TechBlog #Python #OpenCV