From 51543310a2ea5236341d28bcdf2726567b5c1980 Mon Sep 17 00:00:00 2001 From: guojianzhu Date: Sat, 20 Jul 2024 13:31:54 +0800 Subject: [PATCH] fix: v2v more than one face --- src/utils/cropper.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/utils/cropper.py b/src/utils/cropper.py index 8c17bfb..e0e3789 100644 --- a/src/utils/cropper.py +++ b/src/utils/cropper.py @@ -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(