Shafeek Saleem commited on
Commit
2b5e9d0
·
1 Parent(s): 775c70f

face recog fixed

Browse files
Files changed (1) hide show
  1. pages/4_Face Recognition.py +15 -14
pages/4_Face Recognition.py CHANGED
@@ -57,22 +57,23 @@ def step4_page():
57
  st.image(image)
58
  st.info("Select the tolerance level you want for your model! (How much distance between faces to consider it a match. "
59
  "Lower is more strict. 0.6 is typical best performance.)")
60
- tolerance = st.slider('Select tolerance level', 0, 1, 0.6, 0.1)
61
- for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
62
- # See if the face is a match for the known face(s)
63
- matches = face_recognition.compare_faces(known_face_encodings, face_encoding, tolerance=tolerance)
 
64
 
65
- name = "Unknown"
66
- # If a match was found in known_face_encodings, just use the first one.
67
- if True in matches:
68
- first_match_index = matches.index(True)
69
- name = known_face_names[first_match_index]
70
 
71
- face_image = image[top:bottom, left:right]
72
- pil_image = Image.fromarray(face_image)
73
- cols[i].image(pil_image, use_column_width=True)
74
- cols[i].write("Person name: " +name)
75
- i+=1
76
 
77
  st.info("Click on the button below to complete this level!")
78
  if st.button("Complete Level"):
 
57
  st.image(image)
58
  st.info("Select the tolerance level you want for your model! (How much distance between faces to consider it a match. "
59
  "Lower is more strict. 0.6 is typical best performance.)")
60
+ tolerance = st.slider('Select tolerance level', 0.0, 1.0, 0.6, 0.1)
61
+ if tolerance:
62
+ for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
63
+ # See if the face is a match for the known face(s)
64
+ matches = face_recognition.compare_faces(known_face_encodings, face_encoding, tolerance=tolerance)
65
 
66
+ name = "Unknown"
67
+ # If a match was found in known_face_encodings, just use the first one.
68
+ if True in matches:
69
+ first_match_index = matches.index(True)
70
+ name = known_face_names[first_match_index]
71
 
72
+ face_image = image[top:bottom, left:right]
73
+ pil_image = Image.fromarray(face_image)
74
+ cols[i].image(pil_image, use_column_width=True)
75
+ cols[i].write("Person name: " +name)
76
+ i+=1
77
 
78
  st.info("Click on the button below to complete this level!")
79
  if st.button("Complete Level"):