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