OpenSourceJoker/LivePortrait
3
0
Fork 0
mirror of https://github.com/KwaiVGI/LivePortrait.git synced 2025-03-15 14:02:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: image driven, regional control

Browse Source
This commit is contained in:
zhangdingyun 2024-08-19 23:05:16 +08:00
parent d5628afd87
commit 88e5aeb405
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/live_portrait_pipeline.py
View File

@ -131,7 +131,6 @@ class LivePortraitPipeline(object):
# load from video file, AND make motion template # load from video file, AND make motion template
output_fps = int(get_fps(args.driving)) output_fps = int(get_fps(args.driving))
log(f"Load driving video from: {args.driving}, FPS is {output_fps}") log(f"Load driving video from: {args.driving}, FPS is {output_fps}")
driving_rgb_lst = load_video(args.driving) driving_rgb_lst = load_video(args.driving)
elif is_image(args.driving): elif is_image(args.driving):
flag_is_driving_video = False flag_is_driving_video = False
@ -173,6 +172,9 @@ class LivePortraitPipeline(object):
log(f"Dump motion template to {wfp_template}") log(f"Dump motion template to {wfp_template}")
else: else:
raise Exception(f"{args.driving} does not exist!") 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 ######## ######## prepare for pasteback ########
I_p_pstbk_lst = None I_p_pstbk_lst = None