Aadhithya
commited on
Commit
·
4375017
1
Parent(s):
7c1f5ab
Update roop/face_analyser.py
Browse files- 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 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
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]]:
|