Dizi-iris/code_pi.py at main · CyberPanditG/Dizi-iris · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
import cv2 as cv
import RPi.GPIO as GPIO
import time

timer = 0
buzzer = 37

GPIO.setmode(GPIO.BOARD)
GPIO.setup(buzzer,GPIO.OUT)


def set_up():
    global capture
    global eye_cascade
    global face_cascade

    capture = cv.VideoCapture(0)
    eye_cascade = cv.CascadeClassifier('haar_eye.xml')
    face_cascade = cv.CascadeClassifier('haar_face.xml')


def message(frame, message):
    frame = cv.putText(frame, message, (5, frame.shape[0]-10), cv.FONT_HERSHEY_SIMPLEX,
                       1, (0, 0, 255), 2, cv.LINE_AA)


def buzz():
    if(timer > 30):
        print(timer)
        GPIO.output(buzzer,GPIO.LOW)
        time.sleep(0.25)
        GPIO.output(buzzer,GPIO.HIGH)


def rescale_frame(frame, scale=1):
    width = int(frame.shape[1] * scale)
    heigth = int(frame.shape[0] * scale)

    dimensions = (width, heigth)

    return cv.resize(frame, dimensions, interpolation=cv.INTER_AREA)


def eye_tracker(roi_gray, fx, fy, frame):

    global timer

    eye_rect = eye_cascade.detectMultiScale(
        roi_gray, scaleFactor=1.1, minNeighbors=15)

    for (x, y, w, h) in eye_rect:
        x = fx+x
        y = fy+y

        cv.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)

        frame = cv.putText(frame, 'Eye', (x+5, y-5), cv.FONT_HERSHEY_SIMPLEX,
                           .5, (255, 255, 0), 1, cv.LINE_AA)

    if(len(eye_rect) < 1):
        message(frame, 'no eye open')
        timer = timer + 2

    elif(len(eye_rect) == 1):
        message(frame, 'One eye open')
        timer = timer + 1

    elif(len(eye_rect) > 1):
        message(frame, 'Both eyes open')
        timer = 0

    buzz()


def face_tracker(frame):

    gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

    face_rect = face_cascade.detectMultiScale(
        gray, scaleFactor=1.1, minNeighbors=6)

    if(len(face_rect) < 1):
        message(frame, 'No face detected')

    elif(len(face_rect) > 1):
        message(frame, 'only one face...')

    else:
        for (fx, fy, w, h) in face_rect:
            cv.rectangle(frame, (fx, fy), (fx+w, fy+h), (0, 255, 0), 2)
            frame = cv.putText(frame, 'Face', (fx+5, fy-5), cv.FONT_HERSHEY_SIMPLEX,
                               1, (255, 255, 0), 2, cv.LINE_AA)

            roi_gray = gray[fy:fy + h, fx:fx + w]

            eye_tracker(roi_gray, fx, fy, frame)


def main():

    set_up()

    while True:
        isTrue, frame = capture.read()

        face_tracker(frame)

        frame_resized = rescale_frame(frame)
        cv.imshow("video", frame_resized)

        if cv.waitKey(1) & 0xFF == ord('x'):
            break

    capture.release()
    cv.destroyAllWindows()


if __name__ == "__main__":
    main()