Shafeek Saleem commited on
Commit
eceda24
·
1 Parent(s): 3acfa00

added face encodings pagejj

Browse files
Files changed (1) hide show
  1. pages/3_Face Encodings.py +3 -1
pages/3_Face Encodings.py CHANGED
@@ -45,6 +45,8 @@ def step3_page():
45
  )
46
 
47
  img_dir = os.path.join(".sessions", get_login()["username"], "known_faces")
 
 
48
  images = os.listdir(img_dir)
49
  if len(images) > 0:
50
  st.subheader("Let's see your saved faces in your known-face database.")
@@ -62,7 +64,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
- 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!")
 
45
  )
46
 
47
  img_dir = os.path.join(".sessions", get_login()["username"], "known_faces")
48
+ face_encodings_dir = os.path.join(".sessions", get_login()["username"], "face_encodings")
49
+ os.makedirs(face_encodings_dir, exist_ok=True)
50
  images = os.listdir(img_dir)
51
  if len(images) > 0:
52
  st.subheader("Let's see your saved faces in your known-face database.")
 
64
  face_image = face_recognition.load_image_file(os.path.join(img_dir, img))
65
  my_face_encoding = face_recognition.face_encodings(face_image)
66
  my_bar.progress(int((i + 1) / len(images) * 100), text="Generating face encodings...")
67
+ np.save(os.path.join(face_encodings_dir, img.split("_")[1]+".npy"), my_face_encoding)
68
  # face_encodings_dict[img.split("_")[1]] = my_face_encoding.tolist()
69
  my_bar.progress(100, text="Successfully encoded all the known faces!")
70
  st.success("Face encoding completed successfully!")