CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-09-26 17:30:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

Temp fix for shaking face

Browse Source
This commit is contained in:
henryruhs
2023-10-22 22:12:09 +02:00
parent 6e582ac910
commit 5dbc65beed
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/face_analyser.py
+3 -2
View File
@@ -65,12 +65,13 @@ def pre_check() -> bool:
def extract_faces(frame : Frame) -> List[Face]: def extract_faces(frame : Frame) -> List[Face]:
face_detector = get_face_analyser().get('face_detector') face_detector = get_face_analyser().get('face_detector')
faces: List[Face] = [] faces: List[Face] = []
temp_frame = resize_frame_dimension(frame, 640, 640) temp_frame = resize_frame_dimension(frame, 1024, 1024)
temp_frame_height, temp_frame_width, _ = temp_frame.shape temp_frame_height, temp_frame_width, _ = temp_frame.shape
frame_height, frame_width, _ = frame.shape frame_height, frame_width, _ = frame.shape
ratio_height = frame_height / temp_frame_height ratio_height = frame_height / temp_frame_height
ratio_width = frame_width / temp_frame_width ratio_width = frame_width / temp_frame_width
face_detector.setScoreThreshold(0.5) face_detector.setScoreThreshold(0.5)
face_detector.setNMSThreshold(0.5)
face_detector.setTopK(100) face_detector.setTopK(100)
face_detector.setInputSize((temp_frame_width, temp_frame_height)) face_detector.setInputSize((temp_frame_width, temp_frame_height))
with THREAD_SEMAPHORE: with THREAD_SEMAPHORE:
@@ -84,7 +85,7 @@ def extract_faces(frame : Frame) -> List[Face]:
(detection[0:4][0] + detection[0:4][2]) * ratio_width, (detection[0:4][0] + detection[0:4][2]) * ratio_width,
(detection[0:4][1] + detection[0:4][3]) * ratio_height (detection[0:4][1] + detection[0:4][3]) * ratio_height
] ]
kps = (detection[4:14].reshape((5, 2)) * [[ ratio_width, ratio_height ]]).astype(int) kps = (detection[4:14].reshape((5, 2)) * [[ ratio_width, ratio_height ]])
score = detection[14] score = detection[14]
embedding = calc_embedding(frame, kps) embedding = calc_embedding(frame, kps)
normed_embedding = embedding / numpy.linalg.norm(embedding) normed_embedding = embedding / numpy.linalg.norm(embedding)