LuisAbularach · ryank213 · Dec 5, 2018 · Dec 6, 2018 · Dec 6, 2018 · Dec 7, 2018
diff --git a/Gaze Detection/caught/suspect_1.jpg b/Gaze Detection/caught/suspect_1.jpg
diff --git a/Gaze Detection/caught/suspect_10.jpg b/Gaze Detection/caught/suspect_10.jpg
diff --git a/Gaze Detection/caught/suspect_12.jpg b/Gaze Detection/caught/suspect_12.jpg
diff --git a/Gaze Detection/caught/suspect_2.jpg b/Gaze Detection/caught/suspect_2.jpg
diff --git a/Gaze Detection/caught/suspect_3.jpg b/Gaze Detection/caught/suspect_3.jpg
diff --git a/Gaze Detection/caught/suspect_4.jpg b/Gaze Detection/caught/suspect_4.jpg
diff --git a/Gaze Detection/caught/suspect_6.jpg b/Gaze Detection/caught/suspect_6.jpg
diff --git a/Gaze Detection/clean_up.py b/Gaze Detection/clean_up.py
@@ -0,0 +1,51 @@
+import face_recognition
+import os  
+
+
+# Removes duplicates of suspects
+def cleanUp(): 
+    dir = "caught/"
+    fileCount = len(os.listdir(dir))
+    print("num of images: " + str(fileCount))
+    if(os.path.exists(dir + "suspect_1.jpg")):
+        # Adds suspect_1 to the individual suspects array
+        suspect1 = face_recognition.load_image_file(dir + "suspect_1.jpg")
+        suspect1Encoding = face_recognition.face_encodings(suspect1)[0]
+        individual_suspects = [suspect1Encoding]
+        i=2
+        # Iterates from suspect_2 to the last suspect
+        for suspects in range(2,fileCount+1):
+            image,i = loadNextImage(i)
+            try:
+                imageEncoding = face_recognition.face_encodings(image)[0]
+            except IndexError:
+                print("there are no detected faces")
+                i+=1
+                continue        
+            # results is a boolean array that compares the encoding with each individual suspect
+            results = face_recognition.compare_faces(individual_suspects, imageEncoding)
+            if(not True in results):
+                # suspect is a new individual and is added to the array
+                individual_suspects.append(imageEncoding)
+            else:
+                os.remove(dir + "suspect_" + str(i)+ ".jpg")
+                print("suspect_" + str(i) + " was removed")
+            i+=1
+        numIndividualSuspects = len(individual_suspects)
+        print("num of individual suspects: " + str(numIndividualSuspects))
+    else:
+        return
+
+# Gets the next valid image
+# i is the index of the next valid image
+def loadNextImage(j):
+    i=j
+    while os.path.exists(dir):
+        try:
+            image = face_recognition.load_image_file(dir + "suspect_" + str(i) +".jpg")
+            return image,i
+        except FileNotFoundError:
+            i+=1
+            continue
+
+cleanUp()
diff --git a/Gaze Detection/real_time.py b/Gaze Detection/real_time.py
@@ -37,6 +37,14 @@
 solid = True
 initial = True
 
+# Checks if a face encoding exists in current known face encodings
+def checkDupe(face_encoding):
+    results = face_recognition.compare_faces(known_face_encodings, face_encoding)
+    if(not True in results):
+        return False
+    else:
+        return True
+
 #We will only anylyze every 5 frames facial features
 frame_count = 0
 
@@ -106,6 +114,21 @@
                 known_face_encodings.append(suspect_face_encoding)
                 known_face_names.append("SUSPECT_" + str(suspect_num))
 
+                if(len(face_recognition.face_encodings(suspect_image)) != 0):
+                    suspect_face_encoding = face_recognition.face_encodings(suspect_image)[0]
+                    # Removes frame from storage if encoding returns duplicate
+                    if(checkDupe(suspect_face_encoding)):
+                        print("captured frame is a duplicate. Removing...")
+                        os.remove(storage + "/suspect_" + str(suspect_num) + ".jpg")
+                        suspect_num -= 1
+                        continue
+                    #Initialize suspition level
+                    suspicion_levels.append(0)
+                    print("Suspect priorities: "+ str(suspicion_levels))
+                    #add it to list of known faces
+                    known_face_encodings.append(suspect_face_encoding)
+                    known_face_names.append("SUSPECT_" + str(suspect_num))
+                    print("suspect_" + str(suspect_num) + " detected")
 
             face_names.append(name)
 

diff --git a/Gaze Detection/server.html b/Gaze Detection/server.html
@@ -0,0 +1,11 @@
+<!DOCTYPE html>
+<html>
+    <head>
+        <meta charset="utf-8">
+        <title>Gaze Detection Server</title>
+    </head>
+    <body>
+        <h1>Most Recent Suspect</h1>
+        <img src="/caught/suspect_1.jpg" alt="" width="250" height="250" />
+    </body>
+</html>
diff --git a/Gaze Detection/server.py b/Gaze Detection/server.py
@@ -0,0 +1,18 @@
+from http.server import HTTPServer, SimpleHTTPRequestHandler
+
+class Server(SimpleHTTPRequestHandler):
+
+    def do_get(self):
+        if self.path == '/':
+            self.path = '/index.html'
+        try:
+            file_to_open = open(self.path[1:]).read()
+            self.send_response(200)
+        except:
+            file_to_open = "File Not Found"
+            self.send_response(404)
+        self.end_headers()
+        self.wfile.write(bytes(file_to_open, 'utf-8'))
+
+httpd = HTTPServer(('localhost', 8080), Server)
+httpd.serve_forever()