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]]:
|