From 4ed320a86b0cf437af3980ecf0b7b38c92f072a1 Mon Sep 17 00:00:00 2001 From: Ilya Sergeev Date: Thu, 4 Jun 2020 15:39:46 +0300 Subject: [PATCH] fix nan in multiply (#769) Co-authored-by: Ilya Sergeev --- merger/MergeMasked.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/merger/MergeMasked.py b/merger/MergeMasked.py index aaea66a..bb93966 100644 --- a/merger/MergeMasked.py +++ b/merger/MergeMasked.py @@ -232,6 +232,10 @@ def MergeMaskedFace (predictor_func, predictor_input_shape, 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) if ('seamless' in cfg.mode and cfg.color_transfer_mode != 0) or \