From 88e5aeb405038c5e2a154e35ce4304003f74f6ea Mon Sep 17 00:00:00 2001 From: zhangdingyun Date: Mon, 19 Aug 2024 23:05:16 +0800 Subject: [PATCH] feat: image driven, regional control --- src/live_portrait_pipeline.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/live_portrait_pipeline.py b/src/live_portrait_pipeline.py index c738113..396b427 100644 --- a/src/live_portrait_pipeline.py +++ b/src/live_portrait_pipeline.py @@ -131,7 +131,6 @@ class LivePortraitPipeline(object): # load from video file, AND make motion template output_fps = int(get_fps(args.driving)) log(f"Load driving video from: {args.driving}, FPS is {output_fps}") - driving_rgb_lst = load_video(args.driving) elif is_image(args.driving): flag_is_driving_video = False @@ -173,6 +172,9 @@ class LivePortraitPipeline(object): log(f"Dump motion template to {wfp_template}") else: raise Exception(f"{args.driving} does not exist!") + if not flag_is_driving_video: + c_d_eyes_lst = c_d_eyes_lst*n_frames + c_d_lip_lst = c_d_lip_lst*n_frames ######## prepare for pasteback ######## I_p_pstbk_lst = None