From 25324267d68046b885c68cc657ffaf73159b7a74 Mon Sep 17 00:00:00 2001 From: Colombo Date: Sat, 11 Apr 2020 21:33:47 +0400 Subject: [PATCH] =?UTF-8?q?Merger:=20color=20transfer=20=E2=80=98sot-m?= =?UTF-8?q?=E2=80=99=20:=20reduced=20color=20flickering?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- merger/MergeMasked.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/merger/MergeMasked.py b/merger/MergeMasked.py index 74c81da..c59bb1b 100644 --- a/merger/MergeMasked.py +++ b/merger/MergeMasked.py @@ -182,7 +182,7 @@ def MergeMaskedFace (predictor_func, predictor_input_shape, elif cfg.color_transfer_mode == 6: #idt-m prd_face_bgr = imagelib.color_transfer_idt (prd_face_bgr*wrk_face_mask_area_a, dst_face_bgr*wrk_face_mask_area_a) elif cfg.color_transfer_mode == 7: #sot-m - prd_face_bgr = imagelib.color_transfer_sot (prd_face_bgr*wrk_face_mask_area_a, dst_face_bgr*wrk_face_mask_area_a) + prd_face_bgr = imagelib.color_transfer_sot (prd_face_bgr*wrk_face_mask_area_a, dst_face_bgr*wrk_face_mask_area_a, batch_size=15) prd_face_bgr = np.clip (prd_face_bgr, 0.0, 1.0) elif cfg.color_transfer_mode == 8: #mix-m prd_face_bgr = imagelib.color_transfer_mix (prd_face_bgr*wrk_face_mask_area_a, dst_face_bgr*wrk_face_mask_area_a) @@ -256,7 +256,7 @@ def MergeMaskedFace (predictor_func, predictor_input_shape, elif cfg.color_transfer_mode == 6: #idt-m out_face_bgr = imagelib.color_transfer_idt (out_face_bgr*wrk_face_mask_area_a, dst_face_bgr*wrk_face_mask_area_a) elif cfg.color_transfer_mode == 7: #sot-m - out_face_bgr = imagelib.color_transfer_sot (out_face_bgr*wrk_face_mask_area_a, dst_face_bgr*wrk_face_mask_area_a) + out_face_bgr = imagelib.color_transfer_sot (out_face_bgr*wrk_face_mask_area_a, dst_face_bgr*wrk_face_mask_area_a, batch_size=15) out_face_bgr = np.clip (out_face_bgr, 0.0, 1.0) elif cfg.color_transfer_mode == 8: #mix-m out_face_bgr = imagelib.color_transfer_mix (out_face_bgr*wrk_face_mask_area_a, dst_face_bgr*wrk_face_mask_area_a)