Shafeek Saleem commited on
Commit
dd8073b
·
1 Parent(s): 1941e70

bug fixed images - path

Browse files
Files changed (1) hide show
  1. utils/inference.py +7 -6
utils/inference.py CHANGED
@@ -23,12 +23,13 @@ def recognize(image,tolerance):
23
  st.write(distance)
24
  name = 'Unknown'
25
  face_id = 'Unknown'
26
- if True in matches:
27
- match_index = matches.index(True)
28
- name = database[match_index]['name']
29
- face_id = database[match_index]['face_id'].split("_")[1]
30
- distance = round(distance[match_index],2)
31
- cv2.putText(image,str(distance),(left,top-30),cv2.FONT_HERSHEY_SIMPLEX,0.75,(0,255,0),2)
 
32
  cv2.rectangle(image,(left,top),(right,bottom),(0,255,0),2)
33
  cv2.putText(image,name,(left,top-10),cv2.FONT_HERSHEY_SIMPLEX,0.75,(0,255,0),2)
34
  return image, name, face_id
 
23
  st.write(distance)
24
  name = 'Unknown'
25
  face_id = 'Unknown'
26
+ for i in range(len(matches)):
27
+ if matches[i].all():
28
+ match_index = i
29
+ name = database[match_index]['name']
30
+ face_id = database[match_index]['face_id'].split("_")[1]
31
+ distance = round(np.sum(distance[match_index]),2)
32
+ cv2.putText(image,str(distance),(left,top-30),cv2.FONT_HERSHEY_SIMPLEX,0.75,(0,255,0),2)
33
  cv2.rectangle(image,(left,top),(right,bottom),(0,255,0),2)
34
  cv2.putText(image,name,(left,top-10),cv2.FONT_HERSHEY_SIMPLEX,0.75,(0,255,0),2)
35
  return image, name, face_id