[python] 동영상 파일에서 특정 영역의 원근 효과 추가하기

08 Dec 2023

python

이를 위해서는 OpenCV를 사용하여 동영상을 처리하고, 변환 매트릭스(transform matrix)를 사용하여 특정 영역에 원근 효과를 적용합니다.

다음 예시는 OpenCV를 사용하여 동영상 파일에서 사다리꼴 모양의 원근 효과를 적용하는 간단한 방법입니다.

import cv2
import numpy as np

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

# 영상의 가로, 세로 길이
width = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)

# 출력 동영상 파일 설정
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output_video.avi', fourcc, 30.0, (int(width), int(height)))

# 변환 매트릭스 생성
pts1 = np.float32([[100, 100], [300, 100], [200, 200], [200, 400]])
pts2 = np.float32([[100, 100], [300, 100], [100, 200], [300, 400]])
M = cv2.getPerspectiveTransform(pts1, pts2)

# 동영상 프레임 읽기
while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break
    # 원근 효과 적용
    dst = cv2.warpPerspective(frame, M, (int(width), int(height)))
    # 원근 효과가 적용된 프레임 저장
    out.write(dst)

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

이 코드는 ‘input_video.mp4’라는 동영상 파일을 열어서 사다리꼴 모양의 원근 효과를 적용한 후, ‘output_video.avi’라는 파일로 저장하는 예시입니다.

변환 매트릭스를 구성할 때, pts1과 pts2는 각각 변환 전과 변환 후의 좌표를 나타내며, cv2.getPerspectiveTransform() 함수를 사용하여 변환 매트릭스를 얻습니다.

이 코드를 실행하면 동영상 파일에서 특정 영역에 원근 효과를 적용한 새로운 동영상 파일을 얻을 수 있습니다.

OpenCV와 Numpy 라이브러리를 이용하여 동영상 처리와 원근 효과 적용에 대한 더 많은 정보는 아래를 참고하세요.