Spaces:
Sleeping
Sleeping
Update config.yaml
Browse files- 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 |
-
|
15 |
-
eye_ar_thresh: 0.
|
16 |
-
eye_ar_consec_frames:
|
17 |
-
|
18 |
-
#
|
19 |
-
|
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:
|
37 |
-
yawning:
|
38 |
-
head_nod:
|
39 |
-
looking_away:
|
|
|
|
|
|
|
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:
|