[python] OpenCV를 사용하여 이미지에서 동영상에서 동영상 분할하기

22 Nov 2023

python

이미지 처리를 위한 강력한 라이브러리인 OpenCV를 사용하여 동영상에서 동영상을 분할하는 방법에 대해 알아보겠습니다.

동영상에서 이미지 가져오기

먼저, 동영상 파일에서 이미지 프레임을 가져오는 방법을 알아보겠습니다. OpenCV의 VideoCapture 클래스를 사용하여 동영상 파일을 열고, read() 메서드를 호출하여 프레임을 읽어올 수 있습니다.

import cv2

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

# 이미지 프레임 읽어오기
success, frame = video_capture.read()

# 읽어온 프레임이 성공적인지 확인
if success:
    # 이미지 처리 코드 작성
else:
    print("프레임을 읽어올 수 없습니다.")
    
# 동영상 파일 닫기
video_capture.release()

이미지에 대한 동영상에서 동영상 분할하기

이미지를 포함한 동영상 파일에서 동영상을 분할하는 방법을 살펴보겠습니다. OpenCV의 VideoWriter 클래스를 사용하여 분할된 동영상 파일을 생성 및 저장할 수 있습니다.

import cv2

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

# 프레임의 크기와 FPS 정보 가져오기
width = int(video_capture.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(video_capture.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = video_capture.get(cv2.CAP_PROP_FPS)

# 분할된 동영상 파일 저장 설정
output_video = cv2.VideoWriter('output_video.mp4', cv2.VideoWriter_fourcc(*'mp4v'),
                               fps, (width, height))

# 이미지 프레임 읽어오기
success, frame = video_capture.read()

while success:
    # 이미지 처리 코드 작성
    # 분할된 동영상 파일에 프레임 저장
    output_video.write(frame)

    # 다음 프레임 읽어오기
    success, frame = video_capture.read()

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

# 분할된 동영상 파일 저장 종료
output_video.release()

위의 코드에서는 output_video.mp4라는 이름으로 분할된 동영상 파일을 생성하고, 각 프레임을 저장하는 예시입니다.

이제 OpenCV를 사용하여 이미지에서 동영상을 분할하는 방법에 대해 알게되었습니다. 이러한 기술을 사용하면 동영상 처리 및 분석에 활용할 수 있습니다.

참고 자료

OpenCV 공식 문서: https://docs.opencv.org
OpenCV-Python 튜토리얼: https://docs.opencv.org/master/d6/d00/tutorial_py_root.html