From c4ab544fd1001f3a44fb22df6a1a3496c308813c Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Thu, 22 Aug 2024 13:38:48 +0000 Subject: [PATCH] added in tri3d_recolor_lab --- __init__.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/__init__.py b/__init__.py index a7eff72..fbcce61 100644 --- a/__init__.py +++ b/__init__.py @@ -2836,9 +2836,14 @@ class TRI3D_recolor_LAB: mu_2, sigma_2 = get_mu_sigma(array_input=image_2[:, :, i], mask_input=mask_2) + sigma_calculated = sigma_1 * factor_sigma[i] + if factor_sigma[i] < 0: + sigma_calculated = sigma_2 + image_2[:, :, i] = ( ((image_2[:, :, i] - mu_2) / sigma_2) * - (sigma_1 * factor_sigma[i])) + (mu_1 * factor_mean[i]) + sigma_calculated) + (mu_1 * factor_mean[i]) + image_2 = np.clip(image_2, 0, 255) image_2 = image_2.astype(dtype=np.uint8)