Aadhithya commited on
Commit
4375017
·
1 Parent(s): 7c1f5ab

Update roop/face_analyser.py

Browse files
Files changed (1) hide show
  1. roop/face_analyser.py +7 -8
roop/face_analyser.py CHANGED
@@ -26,14 +26,13 @@ def clear_face_analyser() -> Any:
26
  FACE_ANALYSER = None
27
 
28
 
29
- def get_one_face(frame: Frame, position: int = 0) -> Optional[Face]:
30
- many_faces = get_many_faces(frame)
31
- if many_faces:
32
- try:
33
- return many_faces[position]
34
- except IndexError:
35
- return many_faces[-1]
36
- return None
37
 
38
 
39
  def get_many_faces(frame: Frame) -> Optional[List[Face]]:
 
26
  FACE_ANALYSER = None
27
 
28
 
29
+
30
+ def get_one_face(frame: Frame) -> Any:
31
+ face = get_face_analyser().get(frame)
32
+ try:
33
+ return min(face, key=lambda x: x.bbox[0])
34
+ except ValueError:
35
+ return None
 
36
 
37
 
38
  def get_many_faces(frame: Frame) -> Optional[List[Face]]: