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

修复色差问题

Browse Source
This commit is contained in:
Jeremy Bieber 2025-01-13 15:04:37 +08:00 committed by GitHub
parent c0f2dc1a92
commit a1ed7af3e1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/gradio_pipeline.py
View File

@ -23,7 +23,7 @@ from .utils.camera import get_rotation_matrix
from .utils.video import get_fps, has_audio_stream, concat_frames, images2video, add_audio_to_video from .utils.video import get_fps, has_audio_stream, concat_frames, images2video, add_audio_to_video
from .utils.helper import is_square_video, mkdir, dct2device, basename from .utils.helper import is_square_video, mkdir, dct2device, basename
from .utils.retargeting_utils import calc_eye_close_ratio, calc_lip_close_ratio from .utils.retargeting_utils import calc_eye_close_ratio, calc_lip_close_ratio
from skimage.exposure import match_histograms
def update_args(args, user_args): def update_args(args, user_args):
"""update the args according to user inputs """update the args according to user inputs
@ -426,6 +426,8 @@ class GradioPipeline(LivePortraitPipeline):
I_p_lst.append(I_p_i) I_p_lst.append(I_p_i)
if flag_do_crop_input_retargeting_video: if flag_do_crop_input_retargeting_video:
I_p_i = match_histograms(I_p_i,img_crop_256x256_lst[i])
I_p_i = np.clip(I_p_i,0,255).astype(np.uint8)
I_p_pstbk = paste_back(I_p_i, source_M_c2o_lst[i], source_rgb_lst[i], mask_ori_lst[i]) I_p_pstbk = paste_back(I_p_i, source_M_c2o_lst[i], source_rgb_lst[i], mask_ori_lst[i])
I_p_pstbk_lst.append(I_p_pstbk) I_p_pstbk_lst.append(I_p_pstbk)
else: else:
@ -447,6 +449,8 @@ class GradioPipeline(LivePortraitPipeline):
I_p_lst.append(I_p_i) I_p_lst.append(I_p_i)
if flag_do_crop_input_retargeting_video: if flag_do_crop_input_retargeting_video:
I_p_i = match_histograms(I_p_i,img_crop_256x256_lst[i])
I_p_i = np.clip(I_p_i,0,255).astype(np.uint8)
I_p_pstbk = paste_back(I_p_i, source_M_c2o_lst[i], source_rgb_lst[i], mask_ori_lst[i]) I_p_pstbk = paste_back(I_p_i, source_M_c2o_lst[i], source_rgb_lst[i], mask_ori_lst[i])
I_p_pstbk_lst.append(I_p_pstbk) I_p_pstbk_lst.append(I_p_pstbk)