Merge pull request #923 from sarthaxtic/main

Added Drowsiness Detection System

Computer Vision/Drowsiness Detection System/main.py

@@ -0,0 +1,102 @@
+#OpenCV for image processing functions
+import cv2
+# Numpy for array related functions
+import numpy as np
+# Dlib for face landmark detection
+import dlib
+#face_utils for basic operations of conversion
+from imutils import face_utils
+#playsound for playing alarm sound
+from playsound import playsound
+
+#Initializing the camera and taking the instance
+cap = cv2.VideoCapture(0)
+
+#Initializing the face detector and landmark detector
+detector = dlib.get_frontal_face_detector()
+predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")
+
+#status marking for current state
+Sleep = 0
+Drowsy = 0
+Active = 0
+status=""
+color=(0,0,0)
+
+def compute(ptA,ptB):
+	dist = np.linalg.norm(ptA - ptB)
+	return dist
+
+def blink(a,b,c,d,e,f):
+	up = compute(b,d) + compute(c,e)
+	down = compute(a,f)
+	ratio = up/(2.0*down)
+
+	#Checking for blink
+	if(ratio>0.25):
+		return 2
+	elif(ratio>0.21 and ratio<=0.25):
+		return 1
+	else:
+		return 0
+
+
+while True:
+    feature, frame = cap.read()
+    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+
+    faces = detector(gray)
+    face_frame=frame.copy()
+    #detect face in faces array
+    for face in faces:
+        x1 = face.left()
+        y1 = face.top()
+        x2 = face.right()
+        y2 = face.bottom()
+
+        cv2.rectangle(face_frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
+
+        landmarks = predictor(gray, face)
+        landmarks = face_utils.shape_to_np(landmarks)
+
+        #The numbers are actually the landmarks which will show eye
+        left_blink = blink(landmarks[36],landmarks[37], 
+        	landmarks[38], landmarks[41], landmarks[40], landmarks[39])
+        right_blink = blink(landmarks[42],landmarks[43], 
+        	landmarks[44], landmarks[47], landmarks[46], landmarks[45])
+
+        #Now judge what to do for the eye blinks
+        if(left_blink==0 or right_blink==0):
+            Sleep+=1
+            Drowsy=0
+            Active=0
+            if(Sleep>6):
+                playsound('alarm.mp3')
+                status="SLEEPING ALERT!!!"
+                color=(255,0,0)
+        elif(left_blink==1 or right_blink==1):
+            Drowsy+=1
+            Sleep=0
+            Active=0
+            if(Drowsy>6):
+                status="DROWSY"
+                color=(0,0,255)
+        else:
+            Drowsy=0
+            Sleep=0
+            Active+=1
+            if(Active>6):
+                status="ACTIVE :)"
+                color=(0,255,0)
+
+        cv2.putText(frame, status, (100,100), cv2.FONT_HERSHEY_SIMPLEX, 1.2, color,3)
+
+        for n in range(0,68):
+            (x,y)=landmarks[n]
+            cv2.circle(face_frame, (x, y), 1, (255, 255, 255), -1)
+
+    cv2.imshow("Frame",frame)
+    cv2.imshow("Result of detector",face_frame)
+    key=cv2.waitKey(1)
+    if(key==27):
+      	break

Project-Structure.md

@@ -322,6 +322,8 @@
   * Ball-Tracking
     * [Output](Computer%20Vision/Ball-Tracking/Output.ipynb)
     * [Project](Computer%20Vision/Ball-Tracking/Project.ipynb)
+  * Drowsiness Detection System
+    * [Main](Computer%20Vision/Drowsiness%20Detection%20System/main.py)
   * Volume Control Using Hand Gesture
     * [Gesturevolume](Computer%20Vision/Volume%20Control%20Using%20Hand%20Gesture/GestureVolume.py)
     * [Handtrackingmodule](Computer%20Vision/Volume%20Control%20Using%20Hand%20Gesture/HandTrackingModule.py)