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

修复色差问题

Browse Source
This commit is contained in:
Jeremy Bieber 2025-01-13 14:56:26 +08:00 committed by GitHub
parent 1df473f067
commit 3f81c0c8e6
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/live_portrait_pipeline.py
View File

@ -26,7 +26,7 @@ from .utils.filter import smooth
from .utils.rprint import rlog as log from .utils.rprint import rlog as log
# from .utils.viz import viz_lmk # from .utils.viz import viz_lmk
from .live_portrait_wrapper import LivePortraitWrapper from .live_portrait_wrapper import LivePortraitWrapper
from skimage.exposure import match_histograms
def make_abs_path(fn): def make_abs_path(fn):
return osp.join(osp.dirname(osp.realpath(__file__)), fn) return osp.join(osp.dirname(osp.realpath(__file__)), fn)
@ -444,8 +444,12 @@ class LivePortraitPipeline(object):
if inf_cfg.flag_pasteback and inf_cfg.flag_do_crop and inf_cfg.flag_stitching: if inf_cfg.flag_pasteback and inf_cfg.flag_do_crop and inf_cfg.flag_stitching:
# TODO: the paste back procedure is slow, considering optimize it using multi-threading or GPU # TODO: the paste back procedure is slow, considering optimize it using multi-threading or GPU
if flag_is_source_video: if flag_is_source_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_float) I_p_pstbk = paste_back(I_p_i, source_M_c2o_lst[i], source_rgb_lst[i], mask_ori_float)
else: else:
I_p_i = match_histograms(I_p_i,img_crop_256x256)
I_p_i = np.clip(I_p_i,0,255).astype(np.uint8)
I_p_pstbk = paste_back(I_p_i, crop_info['M_c2o'], source_rgb_lst[0], mask_ori_float) I_p_pstbk = paste_back(I_p_i, crop_info['M_c2o'], source_rgb_lst[0], mask_ori_float)
I_p_pstbk_lst.append(I_p_pstbk) I_p_pstbk_lst.append(I_p_pstbk)