Code cleaning: Added some Global Variables

trackmultiplefaces.py

@@ -24,14 +24,27 @@ faceCascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
 OUTPUT_SIZE_WIDTH = 775
 OUTPUT_SIZE_HEIGHT = 600
 
-RED = (0, 0, 255)
-BLUE = (255, 0, 0)
+# Colors
+WHITE  =(255, 255, 255)
+RED    = (0, 0, 255)
+BLUE   = (255, 0, 0)
 PURPLE = (127, 0, 255)
-GREEN = (0, 255, 0)
+GREEN  = (0, 255, 0)
+
 suspiciousColors = {0:(0,0,0), 1:GREEN, 2:PURPLE, 3:RED}
 
 averageSuspicious = 0
 
+NAME_COLOR        = WHITE
+IDENTIFYING_COLOR = WHITE
+
+# The default color of the rectangle we draw around the face
+rectangleColor = (0,165,255)
+rectangleColorSuspicious = RED
+
+# Suspicious threshold
+SUSPICIOUS_THRESHOLD = 2 
+
 #We are not doing really face recognition
 def doRecognizePerson(faceNames, fid):
     time.sleep(2)
@@ -66,9 +79,6 @@ def detectAndTrackMultipleFaces(start_thread,
     #Start the window thread for the two windows we are using
     cv2.startWindowThread()
 
-    #The color of the rectangle we draw around the face
-    rectangleColor = (0,165,255)
-
     #variables holding the current frame number and the current faceid
     frameCounter = 0
     currentFaceID = 0
@@ -287,7 +297,7 @@ def detectAndTrackMultipleFaces(start_thread,
                         (t_x+ int((t_w - len(faceNames[fid])*10)/2 + 10), int(t_y) - 10),
                         cv2.FONT_HERSHEY_SIMPLEX,
                         0.5, 
-                        (255, 255, 255), 
+                        NAME_COLOR, 
                         1, 
                         cv2.LINE_AA)
                     
@@ -296,8 +306,7 @@ def detectAndTrackMultipleFaces(start_thread,
                         put_warning_message(fid, 
                                             faceTrackers, 
                                             faceSuspicion, 
-                                            resultImage, 
-                                            threshold=2)
+                                            resultImage)
                 
                 # Else print Identifying
                 else:
@@ -307,7 +316,7 @@ def detectAndTrackMultipleFaces(start_thread,
                                 (t_x + int((t_w - len(text)*10)/2 + 10), int(t_y)),
                                 cv2.FONT_HERSHEY_SIMPLEX,
                                 0.5, 
-                                (255, 255, 255), 
+                                IDENTIFYING_COLOR, 
                                 1, 
                                 cv2.LINE_AA)
 
@@ -334,7 +343,7 @@ def detectAndTrackMultipleFaces(start_thread,
     cv2.destroyAllWindows()
     exit(0)
 
-def put_warning_message(fid, faceTrackers, faceSuspicion, resultImage, threshold=2):
+def put_warning_message(fid, faceTrackers, faceSuspicion, resultImage):
     '''
     Print a Warning message depending on suspicion level
     '''
@@ -347,8 +356,8 @@ def put_warning_message(fid, faceTrackers, faceSuspicion, resultImage, threshold
     
     suspicion_level = faceSuspicion[fid]
     
-    if suspicion_level > threshold:
-        text = "ATTENTION! INDIVIDU DANGEREUX!"
+    if suspicion_level > SUSPICIOUS_THRESHOLD:
+        text = "ATTENTION! PERSONNE DANGEREUSE!"
         cv2.putText(resultImage, 
                     text ,
                     # Name's alignment
@@ -388,15 +397,15 @@ def rectangle_around_face(fid, faceTrackers, faceSuspicion, draw_person_dangeros
                   person_color,
                   1)
 
-def pick_person_dangerosity_color(suspicionLevel, threshold=2):
+def pick_person_dangerosity_color(suspicionLevel):
     '''
     Compute person level suspicion -> simplified to 2 levels
     Can reintroduce the faceSuspicious dictionnary lated, was more clean
     '''
-    if suspicionLevel> threshold:
-        color = RED
+    if suspicionLevel> SUSPICIOUS_THRESHOLD:
+        color = rectangleColorSuspicious
     else:
-        color = GREEN
+        color = rectangleColor
 
     return color