나도!개발자가 되고싶어

목표

• 파일로부터 비디오 가져오기(cv.VideoCapture)
• 비디오 프레임 이미지 읽기(cv.VideoCapture→read)
• Mat 컬러 변환하기(cv.cvtColor)

소스

import cv2 as cv

cap = cv.VideoCapture('video.mp4')

while True:
    ret, frame = cap.read()

    if not ret:
        print("Can't receive frame (stream end?). Exiting...")
        break

    gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

    cv.imshow('frame', gray)
    if cv.waitKey(25) == ord('q'):
        break

cap.release()
cv.destroyAllWindows()

설명

cap = cv.VideoCapture('video.mp4')

VideoCapture로 파일을 읽습니다.

ret, frame = cap.read()

read()를 하면 처음 리턴 값에 True, False를 반환하는데, 프레임을 읽는게 성공하면 True를 반환하고 실패하면 False를 반환합니다.

그래서 리턴 값으로 마지막 프레임을 확인할 수 있습니다.

gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

cv.cvtColor를 통해 이미지를 흑백으로 변환합니다.