admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove already swapped face from list

Browse Source 
This gives a speedup on my machine, but will have the side effect of preferring faces earlier in the list
This commit is contained in:
Andras Fekete
2023-06-19 10:00:41 -04:00
parent f6c006920d
commit 1388fc651f
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
refacer.py
View File

@@ -155,15 +155,18 @@ class Refacer:
max_num=1 max_num=1
faces = self.__get_faces(frame,max_num=max_num) faces = self.__get_faces(frame,max_num=max_num)
for face in faces: if self.first_face:
if self.first_face: for face in faces:
frame = self.face_swapper.get(frame, face, self.replacement_faces[0][1], paste_back=True) frame = self.face_swapper.get(frame, face, self.replacement_faces[0][1], paste_back=True)
break break
else: else:
for rep_face in self.replacement_faces: for rep_face in self.replacement_faces:
sim = self.rec_app.compute_sim(rep_face[0], face.embedding) for i in range(len(faces) - 1, -1, -1):
sim = self.rec_app.compute_sim(rep_face[0], faces[i].embedding)
if sim>=rep_face[2]: if sim>=rep_face[2]:
frame = self.face_swapper.get(frame, face, rep_face[1], paste_back=True) frame = self.face_swapper.get(frame, faces[i], rep_face[1], paste_back=True)
del faces[i]
break
return frame return frame
def __check_video_has_audio(self,video_path): def __check_video_has_audio(self,video_path):