[5] 동영상 프레임 단위로 나누기

동영상을 프레임 단위로 추출하기 위해 opencv를 사용하였다.

opencv 설치 방법은 pip을 통하여 설치할수 있다.

python -m pip install opencv-python

 

다음과 같이 파이썬 내에서 입력을 하게되면 자신의 opencv 버전이 나오는데 정상적으로 설치가 됐다면 버전이 출력된다.

import cv2
print(cv2.__version__)

 

비디오를 프레임 단위로 자르는 코드는 다음과 같다.

import cv2
vidcap = cv2.VideoCapture('/Your/video/name.h264')
success,image = vidcap.read()

count = 1
success = True

while success:
  success,image = vidcap.read()
  cv2.imwrite("/Your/img/save/path/%d.jpg" % count, image)
  print("saved image %d.jpg" % count)
  
  if cv2.waitKey(10) == 27:                    
      break
  count += 1

 

이렇게 잘라진 이미지를 이용해 yolo 분석에 사용한다.

이번 프로젝트에서는 사용성을 높이기위해 GUI 구현 부분에서 해당 코드를 추가하여 사용했다.

다음과 같이 진행되면서 이미지가 나뉘어 지면 된다.

 

참고

https://www.it-swarm.dev/ko/python/%ED%8C%8C%EC%9D%B4%EC%8D%AC-%EB%B9%84%EB%94%94%EC%98%A4-%ED%94%84%EB%A0%88%EC%9E%84-%EC%B6%94%EC%B6%9C-%EB%B0%8F-%EC%A0%80%EC%9E%A5/1057028285/

python — 파이썬 - 비디오 프레임 추출 및 저장