파이썬과 OpenCV를 사용하여 동영상에서 도형 검출하기

이번 예제에서는 파이썬과 OpenCV를 사용하여 동영상에서 도형을 검출하는 방법을 알아보겠습니다.

준비물

  1. 파이썬 3.x 설치
  2. OpenCV 라이브러리 설치

개요

  1. 동영상 파일 불러오기
  2. 프레임별로 이미지 추출하기
  3. 이미지에서 도형 검출하기
  4. 검출된 도형 표시하기
  5. 결과 동영상 저장하기

코드 예제

import cv2

# 동영상 파일 불러오기
video = cv2.VideoCapture("input.mp4")

# 동영상의 프레임 수 확인하기
frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))

# 동영상 저장을 위한 비디오 라이터 준비
fourcc = cv2.VideoWriter_fourcc(*'XVID')
output = cv2.VideoWriter("output.avi", fourcc, 24.0, (640, 480))

# 각 프레임별로 이미지 추출하기
for i in range(frame_count):
    # 현재 프레임 읽어오기
    ret, frame = video.read()
    
    # 이미지에서 도형 검출하기
    # 검출 알고리즘 적용
    
    # 검출된 도형 표시하기
    # 표시 알고리즘 적용
    
    # 결과 이미지 동영상에 저장하기
    output.write(frame)
    
# 동영상 파일 닫기
video.release()

# 결과 동영상 파일 닫기
output.release()

참고 자료

#python #opencv