[python] OpenCV로 동영상 프레임 추출하기

07 Dec 2023

python

OpenCV는 컴퓨터 비전과 이미지 처리 작업을 위한 라이브러리로, 다양한 기능을 제공합니다. 이번 포스트에서는 OpenCV를 사용하여 동영상의 프레임을 추출하는 방법을 알아보겠습니다.

준비물

Python 3.x
OpenCV 라이브러리
동영상 파일

동영상 프레임 추출하기

다음은 OpenCV를 사용하여 동영상의 프레임을 추출하는 간단한 예제 코드입니다.

import cv2

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

# 프레임 추출할 디렉토리 생성
output_dir = '프레임추출'
os.makedirs(output_dir, exist_ok=True)

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

# 프레임 추출
for frame_num in range(total_frames):
    # 프레임 읽기
    ret, frame = video_capture.read()
    
    # 프레임 추출한 이미지 저장
    cv2.imwrite(f'{output_dir}/프레임{frame_num}.jpg', frame)

# 동영상 파일 닫기
video_capture.release()

위 코드를 실행하면 지정한 동영상 파일에서 각각의 프레임을 추출하여 이미지로 저장합니다. 프레임추출이라는 디렉토리가 생성되고, 각 프레임은 프레임번호.jpg 형식으로 저장됩니다.

마치며

이번 포스트에서는 OpenCV로 동영상의 프레임을 추출하는 방법을 알아보았습니다. 동영상 처리 작업에 활용할 수 있는 OpenCV의 다양한 기능들을 이용하면 보다 다양하고 흥미로운 작업을 할 수 있습니다.

더 많은 OpenCV 관련 정보 및 예제는 OpenCV 공식 문서를 참고하세요.