Shafeek Saleem commited on
Commit
3acfa00
1 Parent(s): 2736c78

added face encodings page dhk

Browse files
pages/3_Face Encodings.py CHANGED
@@ -62,8 +62,7 @@ def step3_page():
62
  face_image = face_recognition.load_image_file(os.path.join(img_dir, img))
63
  my_face_encoding = face_recognition.face_encodings(face_image)
64
  my_bar.progress(int((i + 1) / len(images) * 100), text="Generating face encodings...")
65
- with open(os.path.join(".sessions", get_login()["username"], "face_encodings",img.split("_")[1]+".npy"), 'wb') as f:
66
- np.save(f, my_face_encoding)
67
  # face_encodings_dict[img.split("_")[1]] = my_face_encoding.tolist()
68
  my_bar.progress(100, text="Successfully encoded all the known faces!")
69
  st.success("Face encoding completed successfully!")
 
62
  face_image = face_recognition.load_image_file(os.path.join(img_dir, img))
63
  my_face_encoding = face_recognition.face_encodings(face_image)
64
  my_bar.progress(int((i + 1) / len(images) * 100), text="Generating face encodings...")
65
+ np.save(os.path.join(".sessions", get_login()["username"], "face_encodings",img.split("_")[1]+".npy"), my_face_encoding)
 
66
  # face_encodings_dict[img.split("_")[1]] = my_face_encoding.tolist()
67
  my_bar.progress(100, text="Successfully encoded all the known faces!")
68
  st.success("Face encoding completed successfully!")
pages/4_Face Recognition.py CHANGED
@@ -37,11 +37,10 @@ def step4_page():
37
  known_face_names = []
38
  if len(face_encodings) > 0:
39
  for i, face_encoding in enumerate(face_encodings):
40
- with open(os.path.join(face_encodings_dir, face_encoding), 'rb') as f:
41
- known_face_encoding = np.load(f)
42
- face_name = img.split(".")[0]
43
- known_face_encodings.append(known_face_encoding)
44
- known_face_names.append(face_name)
45
 
46
  while True:
47
  # Grab a single frame of video
 
37
  known_face_names = []
38
  if len(face_encodings) > 0:
39
  for i, face_encoding in enumerate(face_encodings):
40
+ known_face_encoding = np.load(os.path.join(face_encodings_dir, face_encoding))
41
+ face_name = img.split(".")[0]
42
+ known_face_encodings.append(known_face_encoding)
43
+ known_face_names.append(face_name)
 
44
 
45
  while True:
46
  # Grab a single frame of video