From 96161523824bc3d77bf6903028c426458a18b284 Mon Sep 17 00:00:00 2001 From: Jonathan Date: Mon, 21 Mar 2016 22:29:55 -0400 Subject: [PATCH] finished computer vision code --- face_detect.py | 25 ++++++++++++++++++++++++- 1 file changed, 24 insertions(+), 1 deletion(-) diff --git a/face_detect.py b/face_detect.py index 247a4d1..d9d4bea 100644 --- a/face_detect.py +++ b/face_detect.py @@ -1,4 +1,27 @@ """ Experiment with face detection and image filtering using OpenCV """ +import numpy as np import cv2 -import numpy as np \ No newline at end of file + +cap = cv2.VideoCapture(0) +face_cascade = cv2.CascadeClassifier('/home/jon/Toolbox/ToolBox-ComputerVision/haarcascade_frontalface_alt.xml') +kernel = np.ones((25,25),'uint8') + +while(True): + # Capture frame-by-frame + ret, frame = cap.read() + faces = face_cascade.detectMultiScale(frame, scaleFactor=1.2, minSize=(20,20)) + 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.circle(frame,(x+w/2,y+h/2),85,(0,0,0),4) + cv2.circle(frame,(x+w/2-20,y+h/2 - 20),15,(0,0,255),4) + cv2.circle(frame,(x+w/2+20,y+h/2 - 20),15,(0,0,255),4) + cv2.circle(frame,(x+w/2,y+h/2 + 30),30,(255,0,0),4) + # Display the resulting frame + cv2.imshow('frame',frame) + if cv2.waitKey(1) & 0xFF == ord('q'): + break + +# When everything done, release the capture +cap.release() +cv2.destroyAllWindows() \ No newline at end of file