Draw face detection rectangle

ghostjson · Mar 28, 2019 · 1def872 · 1def872
1 parent 966daf7
commit 1def872
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 7 deletions.
diff --git a/core/__pycache__/video.cpython-35.pyc b/core/__pycache__/video.cpython-35.pyc
diff --git a/core/video.py b/core/video.py
@@ -1,5 +1,5 @@
 import cv2
-
+from core.core_face_identification import FaceIdentification
 
 
 def stream():
@@ -13,13 +13,20 @@ def stream():
 
         _,frame = cap.read()   #get each frame
 
-        cv2.imshow('Video',frame) #show each frame
-
-        if not flag:
+
+        #detect face in video
+        face = FaceIdentification.detect_face(frame)
+
+        #draw rec on the frame if detect
+        try:
+            face = face[1]
+            x,y,w,h = face
+            cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
+            cv2.imshow('Video',frame) #show each frame
+
+        except:
+            cv2.imshow('Video',frame) #show each frame
 
-            name = "img/frame.jpg"
-            cv2.imwrite(name,frame)
-            flag = True
         #break when esc key is pressed
         k = cv2.waitKey(1)
         if k == 27:
@@ -31,3 +38,9 @@ def stream():
     cv2.destroyAllWindows()
     cap.release() #release all resources
 
+
+        # if not flag:
+
+            # name = "img/frame.jpg"
+            # cv2.imwrite(name,frame)
+            # flag = True
diff --git a/img/frame.jpg b/img/frame.jpg