Testys commited on
Commit
98c37cb
·
verified ·
1 Parent(s): 817a521

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +15 -25
config.yaml CHANGED
@@ -11,32 +11,22 @@ detection_strategy: "geometric"
11
  # -- Geometric Strategy Settings --
12
  # Parameters for the facial landmark-based detection methods.
13
  geometric_settings:
14
- # Eye Aspect Ratio (EAR) for blink/closure detection
15
- eye_ar_thresh: 0.23
16
- eye_ar_consec_frames: 15
17
-
18
- # Mouth Aspect Ratio (MAR) for yawn detection
19
- yawn_mar_thresh: 0.70
20
- yawn_consec_frames: 20
21
-
22
- # Head Pose Estimation for look-away/nod-off detection
23
- head_nod_thresh: 15.0
24
- head_look_away_thresh: 20.0
25
- head_pose_consec_frames: 20
26
-
27
- # Low Light Detection
28
- low_light_thresh: 70 # Average frame brightness below which a warning is shown (0-255).
29
-
30
- # Drowsiness Level Scoring
31
- # The system will sum the weights of active indicators (eyes, mouth, head).
32
- drowsiness_levels:
33
- very_drowsy_threshold: 0.8 # e.g., Eyes + Head Nod (0.45 + 0.55 = 1.0)
34
- slightly_drowsy_threshold: 0.4 # e.g., Just Yawning (0.30) or Eyes Closed (0.45)
35
  indicator_weights:
36
- eye_closure: 0.45
37
- yawning: 0.30
38
- head_nod: 0.55
39
- looking_away: 0.25
 
 
 
40
 
41
  # -- CNN Model Settings (Not used in 'geometric' mode) --
42
  cnn_model_settings:
 
11
  # -- Geometric Strategy Settings --
12
  # Parameters for the facial landmark-based detection methods.
13
  geometric_settings:
14
+ low_light_thresh: 60
15
+ eye_ar_thresh: 0.24 # Increased: Catches squinting
16
+ eye_ar_consec_frames: 10 # Decreased: Faster eye closure detection
17
+ yawn_mar_thresh: 0.5 # Decreased: Catches smaller yawns
18
+ yawn_consec_frames: 15 # Decreased: Faster yawn detection
19
+ head_nod_thresh: 15.0 # Decreased: Catches smaller nods
20
+ head_look_away_thresh: 18.0 # Decreased: Catches smaller head turns
21
+ head_pose_consec_frames: 10 # Decreased: Faster head pose detection
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  indicator_weights:
23
+ eye_closure: 1
24
+ yawning: 1
25
+ head_nod: 1
26
+ looking_away: 1
27
+ drowsiness_levels:
28
+ slightly_drowsy_threshold: 1 # Unchanged, as it's the minimum
29
+ very_drowsy_threshold: 2 # Could be left at 2, or lowered to 1.5 if using fractional weights
30
 
31
  # -- CNN Model Settings (Not used in 'geometric' mode) --
32
  cnn_model_settings: