1

2020-01-25 18:18:06 +04:00
parent f7f4e44a98
commit a3dfcb91b9
1 changed files with 1 additions and 1 deletions
--- a/core/imagelib/color_transfer.py
+++ b/core/imagelib/color_transfer.py
@@ -299,7 +299,7 @@ def linear_color_transfer(target_img, source_img, mode='pca', eps=1e-5):
    matched_img += mu_s
    matched_img[matched_img>1] = 1
    matched_img[matched_img<0] = 0
-    return matched_img
+    return matched_img.astype(source_img.dtype)

 def lab_image_stats(image):
    # compute the mean and standard deviation of each channel