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
|