Spaces:
Runtime error
Runtime error
Shafeek Saleem
commited on
Commit
·
16b122a
1
Parent(s):
a81dca2
bug fixed - image readnhj
Browse files
pages/2_Face Detection and Creating Database.py
CHANGED
@@ -81,7 +81,7 @@ def step2_page():
|
|
81 |
face_image = image[top:bottom, left:right]
|
82 |
pil_image = Image.fromarray(face_image)
|
83 |
col.image(pil_image, use_column_width=True)
|
84 |
-
face_name = st.text_input('Specify name', "This is a placeholder", key="text_"+str(i))
|
85 |
if st.button("Save", key="button_"+str(i)):
|
86 |
img_name = str(uuid.uuid4()) + f"_{face_name}_{i}" + ".jpg"
|
87 |
img_path = os.path.join(img_dir, img_name)
|
|
|
81 |
face_image = image[top:bottom, left:right]
|
82 |
pil_image = Image.fromarray(face_image)
|
83 |
col.image(pil_image, use_column_width=True)
|
84 |
+
face_name = st.text_input('Specify name to save it in the known-face database', "This is a placeholder", key="text_"+str(i))
|
85 |
if st.button("Save", key="button_"+str(i)):
|
86 |
img_name = str(uuid.uuid4()) + f"_{face_name}_{i}" + ".jpg"
|
87 |
img_path = os.path.join(img_dir, img_name)
|
pages/3_Face Encodings.py
CHANGED
@@ -59,13 +59,9 @@ def step3_page():
|
|
59 |
if len(images) > 0:
|
60 |
for i, img in enumerate(images):
|
61 |
face_image = face_recognition.load_image_file(os.path.join(img_dir, img))
|
62 |
-
|
63 |
-
my_face_encoding_ = face_recognition.face_encodings(face_image)
|
64 |
-
st.write("test 1")
|
65 |
-
my_face_encoding = face_recognition.face_encodings(face_image)[0]
|
66 |
-
st.write("test 2")
|
67 |
my_bar.progress(int((i + 1) / len(images) * 100), text="Generating face encodings...")
|
68 |
-
face_encodings_dict[img.split("_")[1]] = my_face_encoding
|
69 |
my_bar.progress(100, text="Successfully encoded all the known faces!")
|
70 |
st.success("Face encoding completed successfully!")
|
71 |
with open(os.path.join(".sessions", get_login()["username"], "face_encodings.json"), "w") as write_file:
|
|
|
59 |
if len(images) > 0:
|
60 |
for i, img in enumerate(images):
|
61 |
face_image = face_recognition.load_image_file(os.path.join(img_dir, img))
|
62 |
+
my_face_encoding = face_recognition.face_encodings(face_image)
|
|
|
|
|
|
|
|
|
63 |
my_bar.progress(int((i + 1) / len(images) * 100), text="Generating face encodings...")
|
64 |
+
face_encodings_dict[img.split("_")[1]] = my_face_encoding.tolist()
|
65 |
my_bar.progress(100, text="Successfully encoded all the known faces!")
|
66 |
st.success("Face encoding completed successfully!")
|
67 |
with open(os.path.join(".sessions", get_login()["username"], "face_encodings.json"), "w") as write_file:
|