diff --git a/models/ConverterMasked.py b/models/ConverterMasked.py index fee6337..1e52775 100644 --- a/models/ConverterMasked.py +++ b/models/ConverterMasked.py @@ -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()] diff --git a/requirements-cpu.txt b/requirements-cpu.txt index af8677f..40b3833 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -1,4 +1,4 @@ -numpy==1.15.4 +numpy==1.16.1 pathlib==1.0.1 scandir==1.6 h5py==2.7.1 diff --git a/requirements-gpu-cuda9-cudnn7.txt b/requirements-gpu-cuda9-cudnn7.txt index 54df6b8..c54c218 100644 --- a/requirements-gpu-cuda9-cudnn7.txt +++ b/requirements-gpu-cuda9-cudnn7.txt @@ -1,4 +1,4 @@ -numpy==1.15.4 +numpy==1.16.1 pathlib==1.0.1 scandir==1.6 h5py==2.7.1