G-Rost commited on
Commit
83d19be
1 Parent(s): 908bd46

Delete roop/face_analyser.py

Browse files
Files changed (1) hide show
  1. roop/face_analyser.py +0 -34
roop/face_analyser.py DELETED
@@ -1,34 +0,0 @@
1
- import threading
2
- from typing import Any
3
- import insightface
4
-
5
- import roop.globals
6
- from roop.typing import Frame
7
-
8
- FACE_ANALYSER = None
9
- THREAD_LOCK = threading.Lock()
10
-
11
-
12
- def get_face_analyser() -> Any:
13
- global FACE_ANALYSER
14
-
15
- with THREAD_LOCK:
16
- if FACE_ANALYSER is None:
17
- FACE_ANALYSER = insightface.app.FaceAnalysis(name='buffalo_l', providers=roop.globals.execution_providers)
18
- FACE_ANALYSER.prepare(ctx_id=0, det_size=(640, 640))
19
- return FACE_ANALYSER
20
-
21
-
22
- def get_one_face(frame: Frame) -> Any:
23
- face = get_face_analyser().get(frame)
24
- try:
25
- return min(face, key=lambda x: x.bbox[0])
26
- except ValueError:
27
- return None
28
-
29
-
30
- def get_many_faces(frame: Frame) -> Any:
31
- try:
32
- return get_face_analyser().get(frame)
33
- except IndexError:
34
- return None