In [6]:
import cv2
import numpy as np
import scipy.misc
import os
import dlib
import time
os.chdir("/home/mckc/Imagedb/")
import uuid
%matplotlib inline
video_capture = cv2.VideoCapture(0)
predictor_model = "/home/mckc/Downloads/shape_predictor_68_face_landmarks.dat"
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor(predictor_model)
win = dlib.image_window()
cap = cv2.VideoCapture('/home/mckc/Downloads/2.mp4')
In [8]:
i = 0
while(cap.grab()):
# Capture frame-by-frame
# time.sleep(1)
ret, frame = cap.read()
i = i + 1
if i % 9 == 0 :
faces = detector(frame, 1)
win.clear_overlay()
win.set_image(frame)
win.add_overlay(faces)
dlib.hit_enter_to_continue()
# Draw a rectangle around the faces
if len(faces)>0:
for a,b in enumerate(faces):
fac = np.array(frame)[b.top():b.bottom(),b.left():b.right(),:]
scipy.misc.toimage(fac).save(str(uuid.uuid4()) +'.jpg')
# When everything is done, release the capture
video_capture.release()
cv2.destroyAllWindows()
In [ ]: