From ea7a2456c5c633b2ce23c7294333dd1a0526a74a Mon Sep 17 00:00:00 2001 From: iperov Date: Thu, 7 Feb 2019 11:27:43 +0400 Subject: [PATCH] upgrading to numpy==1.16.1, fix ConverterMasked --- models/ConverterMasked.py | 3 ++- requirements-cpu.txt | 2 +- requirements-gpu-cuda9-cudnn7.txt | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) 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