[ch02] Python 기초 사용법 - 실전 코딩 : 동영상 전환 이펙트

동영상 전환 이펙트

• 두 동영상 클립 사이에 추가되는 애니메이션 효과
• 페이드-인(fade-in), 페이드-아웃(fade-out), 디졸브(dissolve), 밀기, 확대 등

구현할 기능

• 두 개의 동영상 동시에 열기
• 첫 번째 동영상의 마지막 N개 프레임과 두 번째 동영상의 처음 N개 프레임을 합성
• 합성된 영상을 동영상으로 저장하기

구현 시물레이션

구현 코드

import sys
import numpy as np
import cv2

# 두 개의 동영상을 열어서 cap1, cap2로 지정
cap1 = cv2.VideoCapture('video1.mp4')
cap2 = cv2.VideoCapture('video2.mp4')

if not cap1.isOpened() or not cap2.isOpend():
   print('video open failed!')
   sys.exit()

# 두 동영상의 크기, FPS는 같다고 가정
frame_cnt1 = round(cap1.get(cv2.CAP_PROP_FRAME_COUNT)
frame_cnt2 = round(cap2.get(cv2.CAP_PROP_FRAME_COUNT)
fps = cap1.get(cv2.CAP_PROP_FPS)
effect_frames = int(fps * 2) # 전환 속도를 결정

print('frame_cnt1:', frame_cnt1)
print('frame_cnt2:', frame_cnt2)
print('FPS:', fps)

delay = int(1000/fps)

w = round(cap1.get(cv2.CAP_PROP_FRAME_WIDTH))
h = round(cap1.get(cv2.CAP_PROP_FRAME_HEIGHT))
fourcc = cv2.VideoWriter_fourcc(*'DIVX')

# 출력 동영상 객체 생성
out = cv2.VideoWriter('output.avi', fourcc, fps, (w, h))

# 1번 동영상
for i in range(frame_cnt1 - effect_frames):
    ret1, frame1 = cap1.read()
    
    if not ret1:
        break
    
    out.write(frame1)
    cv2.imshow('frame', frame1)
    cv2.waitKey(delay)
    
# 합성 과정
for i in range(effect_frames):
    ret1, frame1 = cap1.read()
    ret2, frame2 = cap2.read()
    
    if not ret1 or not ret2:
        print('frame read error!')
        sys.exit()
    
    dx = int(w / effect_frames * i)
    frame_new = np.zeros((h, w, 3), dtype=np.uint8)
    
    frame_new[:, 0:dx, :] = frame2[:, 0:dx, :] # 2번 동영상 먼저 등장
    frame_new[:, dx:w, :] = frame1[:, dx:w, :] # 1번 동영상 점차 사라짐
    
    out.write(frame_new)
    cv2.imshow('frame', frame_new)
    cv2.waitKey(delay)
    
# 2번 동영상
for i in range(effect_frames, frame_cnt2):
    ret2, frame2 = cap2.read()
    
    if not ret2:
        break
        
    out.write(frame2)
    cv2.imshow('frame', frame2)
    cv2.waitKey(delay)
    
cap1.release()
cap2.release()
out.release()
cv2.destroyAllWindows()