__author__ = 'yinchuandong'

import cv2
from face import FaceDetect
model = FaceDetect()

capture = cv2.VideoCapture(0)
size = (int(capture.get(cv2.CAP_PROP_FRAME_WIDTH)),
        int(capture.get(cv2.CAP_PROP_FRAME_HEIGHT)))
#video = cv2.VideoWriter("c:\\1.mp4", cv2.VideoWriter_fourcc('I', '4', '2', '0'), 30, size)
print(capture.isOpened())
num = 0
while True:
    ret, img = capture.read()
    imgArr = model.detect(img)
    for img in imgArr:
        #video.write(img)
        cv2.imshow('Video', img)
        key = cv2.waitKey(1)
        cv2.imwrite('camera/%s.jpg' % (str(num)), img)
        num = num + 1
        if key == ord('q'):
            break

# video.release()
capture.release()
cv2.destroyAllWindows()