파이썬과 OpenCV를 사용하여 동영상에서 도형 검출하기
이번 예제에서는 파이썬과 OpenCV를 사용하여 동영상에서 도형을 검출하는 방법을 알아보겠습니다.
준비물
- 파이썬 3.x 설치
- OpenCV 라이브러리 설치
개요
- 동영상 파일 불러오기
- 프레임별로 이미지 추출하기
- 이미지에서 도형 검출하기
- 검출된 도형 표시하기
- 결과 동영상 저장하기
코드 예제
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()
참고 자료
- OpenCV 공식 문서: https://docs.opencv.org
#python #opencv