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

22 Nov 2023

python

이미지에서 동영상을 추출하는 것은 OpenCV를 사용하여 간단하게 구현할 수 있습니다. OpenCV는 이미지 및 비디오 처리를 위한 강력한 라이브러리로, 다양한 컴퓨터 비전 작업에 사용됩니다.

필요한 패키지 설치하기

먼저, OpenCV를 설치해야 합니다. 아래의 명령어를 사용하여 OpenCV를 설치할 수 있습니다.

pip install opencv-python

이미지에서 동영상 추출하기

이제 이미지에서 동영상을 추출해보겠습니다. 다음은 이미지에서 동영상을 추출하는 간단한 코드입니다.

import cv2

# 이미지 파일 경로
image_path = 'path_to_image.jpg'

# 이미지 읽기
image = cv2.imread(image_path)

# 이미지의 크기 가져오기
height, width, _ = image.shape

# 동영상 저장을 위한 VideoWriter 객체 생성
video_writer = cv2.VideoWriter('output_video.mp4', cv2.VideoWriter_fourcc(*'mp4v'), 30, (width, height))

# 동영상에 이미지를 작성
video_writer.write(image)

# 동영상 저장 종료
video_writer.release()

위의 코드는 path_to_image.jpg 이미지 파일에서 동영상을 추출하여 output_video.mp4 파일로 저장하는 예제입니다. VideoWriter 객체를 생성하여 동영상 스트림을 생성하고, 이미지를 스트림에 작성한 후, 스트림을 저장합니다.

여기서 cv2.VideoWriter_fourcc(*'mp4v')은 동영상 코덱을 지정하는 부분입니다. 여러 동영상 코덱을 사용할 수 있으며, mp4v는 MP4 코덱의 일종입니다. 추출한 동영상의 FPS(frames per second)는 30으로 설정했습니다.

웹캠에서 동영상 추출하기

위의 예제는 이미지 파일에서 동영상을 추출하는 방법을 보여줍니다. 만약 웹캠을 사용하여 실시간으로 동영상을 추출하고 싶다면, 아래와 같이 코드를 수정할 수 있습니다.

import cv2

# 웹캠 열기
capture = cv2.VideoCapture(0)

# 웹캠으로부터 프레임 가져오기
ret, frame = capture.read()

# 프레임의 크기 가져오기
height, width, _ = frame.shape

# 동영상 저장을 위한 VideoWriter 객체 생성
video_writer = cv2.VideoWriter('output_video.mp4', cv2.VideoWriter_fourcc(*'mp4v'), 30, (width, height))

# 프레임을 동영상에 작성
video_writer.write(frame)

# 동영상 저장 종료
video_writer.release()

# 웹캠 해제
capture.release()

위의 코드는 0번 웹캠을 연결하여 실시간으로 프레임을 가져와서 동영상으로 저장하는 예제입니다. 웹캠의 경우에는 VideoCapture 객체를 만들고, capture.read()를 통해 프레임을 읽어옵니다.

마무리

OpenCV를 사용하여 이미지에서 동영상을 추출하는 방법에 대해 알아보았습니다. 이미지 파일이나 웹캠에서 동영상을 추출할 수 있으며, VideoWriter를 사용하여 동영상 스트림을 생성하고 저장할 수 있습니다.

더 많은 OpenCV 기능과 예제 코드를 알아보려면 OpenCV 공식 문서를 참고하세요.

OpenCV 공식 문서: https://docs.opencv.org/

Happy coding!