Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import cv2
- import numpy
- class EyerisDetector:
- def __init__(self):
- self.irises = []
- self.blink_in_previous = False
- self.blinks = 0
- self.font=cv2.FONT_HERSHEY_SIMPLEX
- self.eye_cascade = cv2.CascadeClassifier('haarcascade_eye.xml')
- def run(self):
- capture = cv2.VideoCapture(0)
- k = cv2.waitKey(30) & 0xff
- eye_cascade = cv2.CascadeClassifier('haarcascade_eye.xml')
- while k != 27: # ESC
- ret,frame = capture.read()
- gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
- self.irises=[]
- eyes = self.eye_cascade.detectMultiScale(gray, 1.3, 5)
- irises2 = []
- for (ex, ey, ew, eh) in eyes:
- iris_w = int(ex + float(ew / 2))
- iris_h = int(ey + float(eh / 2))
- irises2.append([numpy.float32(iris_w), numpy.float32(iris_h)])
- self.irises=numpy.array(irises2)
- for w, h in self.irises:
- cv2.circle(frame, (w, h), 2, (0, 0, 255), 2)
- cv2.imshow('Eyeris detector', frame)
- k = cv2.waitKey(30) & 0xff
- old_gray = gray.copy()
- cv2.destroyAllWindows()
- eyeris_detector = EyerisDetector()
- eyeris_detector.run()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement