From b8b9a0251fdb5404537f46fcd9385e5802148713 Mon Sep 17 00:00:00 2001 From: emilykohler Date: Sun, 27 Mar 2016 23:24:20 -0400 Subject: [PATCH] Computer Vision Toolbox --- face_detect.py | 26 +++++++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) diff --git a/face_detect.py b/face_detect.py index 247a4d1..ad38855 100644 --- a/face_detect.py +++ b/face_detect.py @@ -1,4 +1,28 @@ """ Experiment with face detection and image filtering using OpenCV """ import cv2 -import numpy as np \ No newline at end of file +import numpy as np + +capture = cv2.VideoCapture(0) +face_cascade = cv2.CascadeClassifier('/home/emily/ToolBox-ComputerVision/haarcascade_frontalface_alt.xml') +kernel = np.ones((30,30),'uint8') + +while(True): + ret, frame = capture.read() + faces = face_cascade.detectMultiScale(frame, scaleFactor=1.2, minSize=(20,20)) + for (x,y,w,h) in faces: + cv2.rectangle(frame,(x,y),(x+w,y+h),(0,0,0)) + cv2.circle(frame, (x+(w/3), y+(h/3)), w/15, (255, 255, 255), -1) + cv2.circle(frame, (x+(2*w/3), y+(h/3)), w/15, (255, 255, 255), -1) + for (x,y,w,h) in faces: + frame[y:y+h,x:x+w,:] = cv2.dilate(frame[y:y+h,x:x+w,:], kernel) + cv2.rectangle(frame,(x,y),(x+w,y+h),(0,0,0)) + # Display the resulting frame + cv2.imshow('frame',frame) + if cv2.waitKey(1) & 0xFF == ord('q'): + break + +# When everything done, release the capture +capture.release() +cv2.destroyAllWindows +