Spaces:
Runtime error
Runtime error
Testimony Adekoya
commited on
Commit
Β·
26440c6
1
Parent(s):
929f736
Trying again|
Browse files
src/detection/strategies/geometric.py
CHANGED
@@ -31,7 +31,6 @@ class GeometricProcessor(BaseProcessor):
|
|
31 |
brightness = np.mean(gray)
|
32 |
is_low_light = brightness < self.settings['low_light_thresh']
|
33 |
|
34 |
-
ptint(f"Lightning: {is_light_low}")
|
35 |
drowsiness_indicators = {
|
36 |
"drowsiness_level": "Awake", "lighting": "Good", "details": {}
|
37 |
}
|
@@ -84,14 +83,13 @@ class GeometricProcessor(BaseProcessor):
|
|
84 |
levels = self.settings['drowsiness_levels']
|
85 |
if score >= levels['very_drowsy_threshold']: drowsiness_indicators['drowsiness_level'] = "Very Drowsy"
|
86 |
elif score >= levels['slightly_drowsy_threshold']: drowsiness_indicators['drowsiness_level'] = "Slightly Drowsy"
|
87 |
-
|
88 |
drowsiness_indicators['details']['Score'] = score
|
89 |
else: # is_low_light is True
|
90 |
drowsiness_indicators["lighting"] = "Low"
|
91 |
|
92 |
# --- Visualization on Video Frame ---
|
93 |
level = drowsiness_indicators['drowsiness_level']
|
94 |
-
print(level)
|
95 |
score_val = drowsiness_indicators.get("details", {}).get("Score", 0)
|
96 |
color = (0, 255, 0) # Green for Awake
|
97 |
|
@@ -105,7 +103,6 @@ class GeometricProcessor(BaseProcessor):
|
|
105 |
|
106 |
cv2.rectangle(frame, (0, 0), (w, h), color, 10)
|
107 |
status_text = f"Status: {level} (Score: {score_val:.2f})"
|
108 |
-
print(status_text)
|
109 |
cv2.putText(frame, status_text, (20, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2, cv2.LINE_AA)
|
110 |
|
111 |
return frame, drowsiness_indicators, face_landmarks
|
|
|
31 |
brightness = np.mean(gray)
|
32 |
is_low_light = brightness < self.settings['low_light_thresh']
|
33 |
|
|
|
34 |
drowsiness_indicators = {
|
35 |
"drowsiness_level": "Awake", "lighting": "Good", "details": {}
|
36 |
}
|
|
|
83 |
levels = self.settings['drowsiness_levels']
|
84 |
if score >= levels['very_drowsy_threshold']: drowsiness_indicators['drowsiness_level'] = "Very Drowsy"
|
85 |
elif score >= levels['slightly_drowsy_threshold']: drowsiness_indicators['drowsiness_level'] = "Slightly Drowsy"
|
86 |
+
|
87 |
drowsiness_indicators['details']['Score'] = score
|
88 |
else: # is_low_light is True
|
89 |
drowsiness_indicators["lighting"] = "Low"
|
90 |
|
91 |
# --- Visualization on Video Frame ---
|
92 |
level = drowsiness_indicators['drowsiness_level']
|
|
|
93 |
score_val = drowsiness_indicators.get("details", {}).get("Score", 0)
|
94 |
color = (0, 255, 0) # Green for Awake
|
95 |
|
|
|
103 |
|
104 |
cv2.rectangle(frame, (0, 0), (w, h), color, 10)
|
105 |
status_text = f"Status: {level} (Score: {score_val:.2f})"
|
|
|
106 |
cv2.putText(frame, status_text, (20, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2, cv2.LINE_AA)
|
107 |
|
108 |
return frame, drowsiness_indicators, face_landmarks
|