[Python][OpenCV] 웹캠을 사용하여 밝기 변화 시 3초간 반전해서 동영상 저장하기 - 그레이스케일로 저장

🔎 문제

웹캠을 사용하여 회색조(grayscale)로 동영상을 저장하되, 카메라로부터 들어오는 현재 프레임이 직전 프레임보다 이미지 전체의 평균 밝기가 30 넘게 바뀔 경우, 그 시점부터 다음 3초간 반전시켜서 output.avi로 저장해주세요.

* 반전된 부분이 확인될 수 있도록 영상을 촬영해주세요.
* 웹캠 사용이 어려울 경우 스마트폰으로 촬영한 영상을 사용하셔도 좋습니다.
* 전체 영상의 길이는 최소 5초를 넘기고, 최대 15초는 넘지 않도록 조절해주세요.

 

import cv2 as cv

def invert():
    cap = cv.VideoCapture(0) # 웹캠에서 받아오기
    if not cap.isOpened():
        print("Camera open failed!")
        return

    # output.avi 동영상 파일 저장
    w = round(cap.get(cv.CAP_PROP_FRAME_WIDTH))
    h = round(cap.get(cv.CAP_PROP_FRAME_HEIGHT))
    fps = cap.get(cv.CAP_PROP_FPS) # 저장에 필요한 정보들
    # fps는 1초 당 프레임 개수

    fourcc = cv.VideoWriter_fourcc(*'DIVX')
    delay = round(1000 / fps)

    outputVideo = cv.VideoWriter('output.avi', fourcc, fps, (w, h), isColor=False) # VideoWriter 객체초기화
    if not outputVideo.isOpened():
        print('File open failed!')
        return

    previous = None
    pre_frame = None
    frame_cnt = -1

    # 프레임정보 가져와서 저장
    while True:
        ret, frame = cap.read() # 각 프레임 읽기
        if not ret:
            break

        if pre_frame is not None:
            previous = pre_frame.mean() # 직전 프레임의 평균 밝기
        current = frame.mean() # 현재 프레임의 평균 밝기

        gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY) # 그레이스케일로 변환
        inversed = ~gray

        # 현재 프레임이 직전 프레임보다 평균 밝기가 30이상 바뀔경우 3초간 반전
        if previous is not None and ((previous+30 <= current)or(previous-30 >= current)):
            frame_cnt = 0 # 밝기 30 변화시 frame_cnt를 0으로

        # 3초간 반전 -> 3 * fps 개의 프레임을 반전하기 (fps가 1초당 프레임 수 이므로)
        if frame_cnt == 3 * fps: # 3초간 반전 완료 시 frame_cnt를 다시 -1로 해놓기
            frame_cnt = -1

        if frame_cnt != -1: # 밝기가 30이상 차이나는 경우
            outputVideo.write(inversed) # 반전된 영상 저장
            cv.imshow('frame', inversed)
            frame_cnt += 1
        else:
            outputVideo.write(gray) # 원본 저장
            cv.imshow('frame', gray)

        # cv.imshow('gray frame', gray)  # 원본 화면에 출력
        # cv.imshow('inversed', inversed)  # 반전 화면에 출력

        if cv.waitKey(delay) == 27:
            break

        pre_frame = frame # 직전 프레임 정보 저장

    cv.destroyAllWindows()

invert()

 

output.avi