Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,82 +1,104 @@
|
|
1 |
-
import
|
2 |
-
import numpy as np
|
3 |
-
import
|
4 |
-
import
|
5 |
-
import
|
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
import numpy as np
|
3 |
+
import cv2 as cv
|
4 |
+
from tensorflow.keras.preprocessing.image import img_to_array
|
5 |
+
import pygame
|
6 |
+
import time
|
7 |
+
|
8 |
+
# Initialize alarm
|
9 |
+
pygame.mixer.init()
|
10 |
+
pygame.mixer.music.load("alarm.wav")
|
11 |
+
|
12 |
+
# Load the pre-trained CNN model for drowsiness detection
|
13 |
+
#model = load_model('drowiness_new6.h5')
|
14 |
+
|
15 |
+
# Define labels
|
16 |
+
labels = ['Closed', 'Open']
|
17 |
+
|
18 |
+
# Load face and eye cascade classifiers
|
19 |
+
face_cascade = cv.CascadeClassifier(cv.data.haarcascades + "haarcascade_frontalface_default.xml")
|
20 |
+
eye_cascade = cv.CascadeClassifier(cv.data.haarcascades + "haarcascade_eye.xml")
|
21 |
+
|
22 |
+
st.title("Real-time Eye Detection and Drowsiness Alert")
|
23 |
+
st.markdown("**Close your eyes for 3 seconds to trigger the alarm.**")
|
24 |
+
|
25 |
+
|
26 |
+
# Initialize webcam
|
27 |
+
cam = cv.VideoCapture(0)
|
28 |
+
FRAME_WINDOW = st.image([])
|
29 |
+
run = st.button("Start")
|
30 |
+
stop = st.button("Stop")
|
31 |
+
|
32 |
+
# Variables to track eye closure
|
33 |
+
drowsiness_start_time = None
|
34 |
+
alarm_triggered = False
|
35 |
+
|
36 |
+
# Function to preprocess the frame for model input
|
37 |
+
def preprocess_frame(frame):
|
38 |
+
frame = cv.resize(frame, (24, 24)) # Resize to match model input shape
|
39 |
+
frame = frame.astype("float") / 255.0 # Normalize
|
40 |
+
frame = img_to_array(frame)
|
41 |
+
frame = np.expand_dims(frame, axis=0)
|
42 |
+
return frame
|
43 |
+
|
44 |
+
running = False
|
45 |
+
|
46 |
+
if run:
|
47 |
+
running = True
|
48 |
+
|
49 |
+
while running:
|
50 |
+
ret, frame = cam.read()
|
51 |
+
if not ret:
|
52 |
+
st.warning("Failed to access webcam.")
|
53 |
+
break
|
54 |
+
|
55 |
+
frame = cv.flip(frame, 1)
|
56 |
+
gray_frame = cv.cvtColor(frame, cv.COLOR_BGR2GRAY) # Convert to grayscale
|
57 |
+
|
58 |
+
# Detect faces
|
59 |
+
faces = face_cascade.detectMultiScale(gray_frame, scaleFactor=1.1, minNeighbors=5)
|
60 |
+
eye_status = 'Closed'
|
61 |
+
|
62 |
+
for (x, y, w, h) in faces:
|
63 |
+
face_roi = frame[y:y+h, x:x+w]
|
64 |
+
roi_gray = gray_frame[y:y+h, x:x+w]
|
65 |
+
eyes = eye_cascade.detectMultiScale(roi_gray, scaleFactor=1.1, minNeighbors=5)
|
66 |
+
|
67 |
+
# Draw rectangle around face
|
68 |
+
cv.rectangle(frame, (x, y), (x+w, y+h), (255, 0, 0), 2)
|
69 |
+
|
70 |
+
# Draw ellipses around detected eyes
|
71 |
+
if len(eyes) > 0:
|
72 |
+
eye_status = 'Open'
|
73 |
+
for (ex, ey, ew, eh) in eyes:
|
74 |
+
center = (x + ex + ew // 2, y + ey + eh // 2)
|
75 |
+
axes = (ew // 2, eh // 2)
|
76 |
+
cv.ellipse(frame, center, axes, 0, 0, 360, (0, 255, 0), 2)
|
77 |
+
drowsiness_start_time = None
|
78 |
+
if alarm_triggered:
|
79 |
+
pygame.mixer.music.stop()
|
80 |
+
alarm_triggered = False
|
81 |
+
else:
|
82 |
+
if drowsiness_start_time is None:
|
83 |
+
drowsiness_start_time = time.time()
|
84 |
+
elif time.time() - drowsiness_start_time >= 2:
|
85 |
+
if not alarm_triggered:
|
86 |
+
pygame.mixer.music.play()
|
87 |
+
alarm_triggered = True
|
88 |
+
|
89 |
+
# Display eye status text
|
90 |
+
cv.putText(frame, f'Eyes: {eye_status}', (30, 30), cv.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
|
91 |
+
|
92 |
+
frame = cv.cvtColor(frame, cv.COLOR_BGR2RGB)
|
93 |
+
FRAME_WINDOW.image(frame)
|
94 |
+
|
95 |
+
# Stop condition
|
96 |
+
if stop:
|
97 |
+
running = False
|
98 |
+
cam.release()
|
99 |
+
cv.destroyAllWindows()
|
100 |
+
pygame.mixer.quit()
|
101 |
+
break
|
102 |
+
|
103 |
+
cam.release()
|
104 |
+
cv.destroyAllWindows()
|