OpenSourceJoker/LivePortrait
3
0
Fork 0
mirror of https://github.com/KwaiVGI/LivePortrait.git synced 2024-12-22 20:42:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: v2v more than one face

Browse Source
This commit is contained in:
guojianzhu 2024-07-20 13:31:54 +08:00
parent 3cb20ffddb
commit 51543310a2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/utils/cropper.py
View File

@ -117,9 +117,10 @@ class Cropper(object):
return ret_dct
def crop_source_video(self, source_rgb_lst, crop_cfg: CropConfig):
def crop_source_video(self, source_rgb_lst, crop_cfg: CropConfig, **kwargs):
"""Tracking based landmarks/alignment and cropping"""
trajectory = Trajectory()
direction = kwargs.get("direction", "large-small")
for idx, frame_rgb in enumerate(source_rgb_lst):
if idx == 0 or trajectory.start == -1:
src_face = self.face_analysis_wrapper.get(