[python] OpenCV를 사용하여 이미지에서 동영상 플레이백 제어하기

22 Nov 2023

python

OpenCV는 영상 처리를 위한 파이썬 라이브러리로서, 이미지와 동영상을 다루는 다양한 기능들을 제공합니다. 이번 블로그 포스트에서는 OpenCV를 사용하여 이미지에서 동영상 플레이백을 제어하는 방법에 대해 알아보겠습니다.

동영상 파일 열기

먼저, 동영상 파일을 열기 위해 OpenCV의 VideoCapture 클래스를 사용합니다. 이 클래스를 사용하면 동영상 파일을 열고 플레이백을 제어할 수 있습니다. 아래는 동영상 파일을 열고 해당 객체를 생성하는 코드입니다.

import cv2

# 동영상 파일 열기
video = cv2.VideoCapture('video.mp4')

동영상 플레이백 제어

동영상 파일을 열었다면, 이제 플레이백을 제어할 수 있습니다. VideoCapture 객체의 read() 함수를 사용하여 프레임 단위로 동영상을 읽으며, imshow() 함수를 사용하여 읽은 프레임을 화면에 표시합니다. 아래는 동영상을 플레이백하는 코드입니다.

import cv2

# 동영상 파일 열기
video = cv2.VideoCapture('video.mp4')

while True:
    # 프레임 단위로 동영상 읽기
    ret, frame = video.read()
    
    # 동영상이 끝날 경우, 종료
    if not ret:
        break
    
    # 읽은 프레임을 화면에 표시
    cv2.imshow('Video', frame)
    
    # 'q' 키를 누르면 플레이백 종료
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

video.release()
cv2.destroyAllWindows()

플레이백 제어 기능 추가

동영상 플레이백을 보다 세밀하게 제어하기 위해 몇 가지 기능을 추가해볼 수 있습니다. 아래는 동영상 플레이백을 제어하는 기능을 추가한 코드입니다.

import cv2

# 동영상 파일 열기
video = cv2.VideoCapture('video.mp4')

# 동영상의 총 프레임 수 확인
total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))

# 플레이백 속도 설정
fps = 30

while True:
    # 프레임 단위로 동영상 읽기
    ret, frame = video.read()
    
    # 동영상이 끝날 경우, 종료
    if not ret:
        break
    
    # 읽은 프레임을 화면에 표시
    cv2.imshow('Video', frame)
    
    # 플레이백 속도 조절
    delay = int(1000 / fps)
    if cv2.waitKey(delay) & 0xFF == ord('q'):
        break
    
    # 플레이백 위치 표시
    current_frame = int(video.get(cv2.CAP_PROP_POS_FRAMES))
    progress = (current_frame / total_frames) * 100
    print(f'플레이백 위치: {progress}%')

video.release()
cv2.destroyAllWindows()

위 코드에서는 cv2.CAP_PROP_FRAME_COUNT 속성을 사용하여 동영상의 총 프레임 수를 가져옵니다. 또한, fps 변수를 설정하여 플레이백 속도를 조절할 수 있습니다. cv2.CAP_PROP_POS_FRAMES 속성을 사용하여 현재 플레이백 위치를 가져온 뒤, 이를 이용하여 플레이백 위치를 표시합니다.

이렇게 OpenCV를 사용하여 이미지에서 동영상 플레이백을 제어할 수 있습니다. OpenCV는 다양한 기능을 제공하므로, 필요한 기능을 활용하여 원하는 동작을 구현할 수 있습니다.

참고 자료

OpenCV 공식 문서