Testys commited on
Commit
2a23c85
·
verified ·
1 Parent(s): 704417d

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