[GDSC] OpenCV로 비디오 read 🎥

카메라로 비디오 읽기

import cv2

cap = cv2.VideoCapture(0)
if cap.isOpen():
	print('width: {}, height : {}'.format(cap.get(3), cap.get(4))
	
while True:
	ret, fram = cap.read()
	
	if ret:
		gray = cv2.cvtColor(fram, cv2.COLOR_BGR2GRAY)
		cv2.imshow('video', gray)
		k == cv2.waitKey(1) & 0xFF
		if k == 27:
			break
	else:
		print('error')

cap.release()
cv2.destroyAllWindows()	

cv2.VideoCapture()를 사용해 비디오 캡쳐 객체 생성 가능
()안의 숫자는 장치 인덱스(어떤 카메라를 사용할 것인가)

• 1개만 부착되어 있으면 0
• 2개 이상이면 첫 웹캠은 0 두번째 웹캠은 1로

cap.isOpen() : 비디오 캡쳐 객체가 정상적으로 Open되었는지 확인

while True: 특정 키를 누를때까지 무한 반복을 위해 사용
ret, fram = cap.read() : 비디오의 한 프레임씩 read
제대로 프레임을 읽으면 ret값이 True, 실패하면 False
fram에 읽은 프레임이 나옴

cv2.cvtColor(): frame을 흑백으로 변환

나는 위와 같이 COLOR_BGR2GRAY로 처리할 예정 !

cap.release() : 오픈한 캡쳐 객체를 해제

카메라로부터 비디오 읽기

import cv2

cap = cv2.VideoCapture('vtest.avi')

while(cap.isOpened()):
    ret, frame = cap.read()
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    cv2.imshow('frame',gray)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
cap.release()
cv2.destroyAllWindows()

비디오 저장

import cv2

cap = cv2.VideoCapture(0)

fourcc = cv2.VideoWriter_fourcc(*'DIVX')
out = cv2.VideoWriter('output.avi', fourcc, 25.0, (640,480))

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

    if ret:
        # 이미지 반전,  0:상하, 1 : 좌우
        frame = cv2.flip(frame, 0)

        out.write(frame)

        cv2.imshow('frame', frame)

        if cv2.waitKey(0) & 0xFF == ord('q'):
            break
    else:
        break

cap.release()
out.release()
cv2.destroyAllWindows()

cv2.VideoWriter(outputFile, fourcc, frame, size) : fourcc는 코덱 정보, frame은 초당 저장될 프레임, size는 저장될 사이즈

Reference