github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-15 01:23:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

upgrading to numpy==1.16.1, fix ConverterMasked

Browse source
This commit is contained in:
iperov 2019-02-07 11:27:43 +04:00
parent 0c1638e168
commit ea7a2456c5
3 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/ConverterMasked.py
View file

@ -198,7 +198,8 @@ class ConverterMasked(ConverterBase):
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 )
img_prd_hborder_rect_mask_a = np.expand_dims (img_prd_hborder_rect_mask_a, -1)
img_mask_blurry_aaa *= img_prd_hborder_rect_mask_a
img_mask_blurry_aaa = np.clip( img_mask_blurry_aaa, 0, 1.0 )
if debug:
debugs += [img_mask_blurry_aaa.copy()]