github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 14:24:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix nan in multiply

Browse source
This commit is contained in:
Ilya Sergeev 2020-06-04 15:30:54 +03:00
commit 0f0d8dd407
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
merger/MergeMasked.py
View file

@ -232,6 +232,10 @@ def MergeMaskedFace (predictor_func, predictor_input_shape,
cfg_mp = cfg.motion_blur_power / 100.0 cfg_mp = cfg.motion_blur_power / 100.0
# linux opencv can produce nan's so there will be errors in multiplying and glitches in videos
img_bgr = np.nan_to_num(img_bgr)
img_face_mask_a = np.nan_to_num(img_face_mask_a)
out_img = np.nan_to_num(out_img)
out_img = img_bgr*(1-img_face_mask_a) + (out_img*img_face_mask_a) out_img = img_bgr*(1-img_face_mask_a) + (out_img*img_face_mask_a)
if ('seamless' in cfg.mode and cfg.color_transfer_mode != 0) or \ if ('seamless' in cfg.mode and cfg.color_transfer_mode != 0) or \