nodes/shadowcz007-comfyui-liveportrait
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update live_portrait_pipeline.py

Browse Source
This commit is contained in:
shadowcz007
2024-07-06 13:56:15 +08:00
parent 8e6a81c5ef
commit 2b33e8cc8c
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/LivePortrait/src/live_portrait_pipeline.py
+5 -3
View File
@@ -190,17 +190,19 @@ class LivePortraitPipeline(object):
if not os.path.exists(directory):
mkdir(directory)
wfp_concat = args.output_path_concat
video_fps = cv2.VideoCapture(args.driving_info).get(cv2.CAP_PROP_FPS)
if is_video(args.driving_info):
frames_concatenated = concat_frames(I_p_lst, driving_rgb_lst, img_crop_256x256)
# save (driving frames, source image, drived frames) result
# wfp_concat = osp.join(directory, f'{basename(args.source_image)}--{basename(args.driving_info)}_concat.mp4')
images2video(frames_concatenated, wfp=wfp_concat)
# images2video(frames_concatenated, wfp=wfp_concat)
images2video(frames_concatenated, wfp=wfp_concat, fps=video_fps)
# save drived result
wfp = args.output_path
if inference_cfg.flag_pasteback:
images2video(I_p_paste_lst, wfp=wfp)
images2video(I_p_paste_lst, wfp=wfp, fps=video_fps)
else:
images2video(I_p_lst, wfp=wfp)
images2video(I_p_lst, wfp=wfp, fps=video_fps)
return wfp, wfp_concat