[python] 파이썬 moviepy로 동영상 크로마키 처리하기

동영상 편집을 위한 파이썬 라이브러리인 moviepy는 강력한 기능을 제공합니다. 이 중에서도 동영상에서 크로마키(색상 키) 처리를 수행하는 방법을 살펴보겠습니다. 크로마키 처리는 배경색을 제거하고 원하는 배경 이미지를 추가하는 데 사용됩니다.

1. moviepy 설치하기

먼저 moviepy를 설치해야 합니다. 다음 명령을 사용하여 moviepy를 설치할 수 있습니다:

pip install moviepy

2. 크로마키 처리하기

크로마키 처리를 하기 위해서는 주어진 동영상에서 특정 색을 제거해야 합니다. 이를 위해 다음과 같은 단계를 따릅니다:

  1. 크로마키로 사용할 색상 값의 범위를 설정합니다.
  2. 동영상에서 특정 프레임을 선택합니다.
  3. 선택한 프레임에서 크로마키로 설정한 색상에 해당하는 픽셀을 제거합니다.
  4. 크로마키 처리된 프레임을 동영상에 삽입합니다.

아래는 파이썬 코드를 사용하여 크로마키 처리를 수행하는 예시입니다:

import moviepy.editor as mp

# 크로마키로 사용할 색상 범위 설정
chroma_key = (0, 255, 0)  # 예시로 초록색을 사용

# 동영상 불러오기
video = mp.VideoFileClip("input.mov")

# 각 프레임에 대해 크로마키 처리 수행
processed_frames = []
for frame in video.iter_frames():
    # 크로마키 색상 제거
    mask = (frame[:,:,0] == chroma_key[0]) & (frame[:,:,1] == chroma_key[1]) & (frame[:,:,2] == chroma_key[2])
    processed_frame = frame.copy()
    processed_frame[mask] = [0, 0, 0]  # 배경 제거

    processed_frames.append(processed_frame)

# 처리된 프레임들로 새로운 동영상 생성
processed_video = mp.ImageSequenceClip(processed_frames, fps=video.fps)

# 출력 동영상 저장
processed_video.write_videofile("output.mp4")

위의 예제에서는 주어진 동영상에서 초록색 배경을 제거하고, 투명한 배경으로 처리한 후 “output.mp4”라는 이름으로 새로운 동영상을 저장합니다.

마무리

moviepy를 사용하면 파이썬으로 동영상 편집 작업을 간단하게 수행할 수 있습니다. 이번 포스트에서는 moviepy를 사용하여 동영상 크로마키 처리하는 방법을 살펴보았습니다. 다양한 기능들을 활용하여 원하는 동영상 편집 작업을 수행해보세요.

참고 자료: