ckcl commited on
Commit
6b8a075
·
verified ·
1 Parent(s): ac9516f

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +40 -0
README.md CHANGED
@@ -1,3 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # Driver Drowsiness Detection Model
2
 
3
  This model is designed to detect driver drowsiness from facial images using a CNN architecture.
@@ -32,3 +50,25 @@ is_drowsy = prediction[0][0] > 0.5
32
  - Training method: Binary cross-entropy loss with Adam optimizer
33
  - Validation split: 20%
34
  - Early stopping with patience=3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: mit
4
+ library_name: tensorflow
5
+ tags:
6
+ - computer-vision
7
+ - drowsiness-detection
8
+ - driver-safety
9
+ - cnn
10
+ - tensorflow
11
+ datasets:
12
+ - custom
13
+ metrics:
14
+ - accuracy
15
+ - binary-crossentropy
16
+ pipeline_tag: image-classification
17
+ ---
18
+
19
  # Driver Drowsiness Detection Model
20
 
21
  This model is designed to detect driver drowsiness from facial images using a CNN architecture.
 
50
  - Training method: Binary cross-entropy loss with Adam optimizer
51
  - Validation split: 20%
52
  - Early stopping with patience=3
53
+
54
+ ## Model Architecture
55
+ - Input Layer: 64x64x3 images
56
+ - Convolutional Layers:
57
+ - Conv2D(32, 3x3) + BatchNorm + ReLU
58
+ - MaxPooling2D(2x2)
59
+ - Conv2D(64, 3x3) + BatchNorm + ReLU
60
+ - MaxPooling2D(2x2)
61
+ - Conv2D(128, 3x3) + BatchNorm + ReLU
62
+ - MaxPooling2D(2x2)
63
+ - Dense Layers:
64
+ - Dense(128) + BatchNorm + ReLU
65
+ - Dropout(0.5)
66
+ - Dense(1) + Sigmoid
67
+
68
+ ## Performance
69
+ - Binary classification for drowsiness detection
70
+ - Optimized for real-time inference
71
+ - Suitable for embedded systems and edge devices
72
+
73
+ ## License
74
+ This model is released under the MIT License.