Spaces:
Sleeping
Sleeping
Update src/detection/strategies/geometric.py
Browse files
src/detection/strategies/geometric.py
CHANGED
@@ -31,6 +31,7 @@ class GeometricProcessor(BaseProcessor):
|
|
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 |
}
|
@@ -80,11 +81,12 @@ class GeometricProcessor(BaseProcessor):
|
|
80 |
levels = self.settings['drowsiness_levels']
|
81 |
if score >= levels['very_drowsy_threshold']: drowsiness_indicators['drowsiness_level'] = "Very Drowsy"
|
82 |
elif score >= levels['slightly_drowsy_threshold']: drowsiness_indicators['drowsiness_level'] = "Slightly Drowsy"
|
83 |
-
|
84 |
drowsiness_indicators['details']['Score'] = score
|
85 |
|
86 |
# --- Visualization on Video Frame ---
|
87 |
level = drowsiness_indicators['drowsiness_level']
|
|
|
88 |
score_val = drowsiness_indicators.get("details", {}).get("Score", 0)
|
89 |
color = (0, 255, 0) # Green for Awake
|
90 |
|
@@ -101,6 +103,7 @@ class GeometricProcessor(BaseProcessor):
|
|
101 |
|
102 |
# Display status text
|
103 |
status_text = f"Status: {level} (Score: {score_val:.2f})"
|
|
|
104 |
cv2.putText(frame, status_text, (20, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2, cv2.LINE_AA)
|
105 |
|
106 |
return frame, drowsiness_indicators, face_landmarks
|
|
|
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 |
}
|
|
|
81 |
levels = self.settings['drowsiness_levels']
|
82 |
if score >= levels['very_drowsy_threshold']: drowsiness_indicators['drowsiness_level'] = "Very Drowsy"
|
83 |
elif score >= levels['slightly_drowsy_threshold']: drowsiness_indicators['drowsiness_level'] = "Slightly Drowsy"
|
84 |
+
|
85 |
drowsiness_indicators['details']['Score'] = score
|
86 |
|
87 |
# --- Visualization on Video Frame ---
|
88 |
level = drowsiness_indicators['drowsiness_level']
|
89 |
+
print(level)
|
90 |
score_val = drowsiness_indicators.get("details", {}).get("Score", 0)
|
91 |
color = (0, 255, 0) # Green for Awake
|
92 |
|
|
|
103 |
|
104 |
# Display status text
|
105 |
status_text = f"Status: {level} (Score: {score_val:.2f})"
|
106 |
+
print(status_text)
|
107 |
cv2.putText(frame, status_text, (20, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2, cv2.LINE_AA)
|
108 |
|
109 |
return frame, drowsiness_indicators, face_landmarks
|