From d16759bcd3d3b8f3eedc83cf81f0dd4257b688c7 Mon Sep 17 00:00:00 2001 From: iperov Date: Sun, 11 Aug 2019 13:22:01 +0400 Subject: [PATCH] ConverterMasked: added mask gradient of bottom area, same as side gradient --- converters/ConverterMasked.py | 1 + 1 file changed, 1 insertion(+) diff --git a/converters/ConverterMasked.py b/converters/ConverterMasked.py index 34e494c..d080e49 100644 --- a/converters/ConverterMasked.py +++ b/converters/ConverterMasked.py @@ -260,6 +260,7 @@ class ConverterMasked(Converter): prd_border_size = int ( prd_hborder_rect_mask_a.shape[1] * self.clip_hborder_mask_per ) prd_hborder_rect_mask_a[:,0:prd_border_size,:] = 0 prd_hborder_rect_mask_a[:,-prd_border_size:,:] = 0 + prd_hborder_rect_mask_a[-prd_border_size:,:,:] = 0 prd_hborder_rect_mask_a = np.expand_dims(cv2.blur(prd_hborder_rect_mask_a, (prd_border_size, prd_border_size) ),-1) img_prd_hborder_rect_mask_a = cv2.warpAffine( prd_hborder_rect_mask_a, face_output_mat, img_size, np.zeros(img_bgr.shape, dtype=np.float32), cv2.WARP_INVERSE_MAP | cv2.INTER_LANCZOS4 )