From 3ae562189dbf2f585dd261b1993fdedddc6eaadb Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Tue, 13 Aug 2019 00:53:47 -0700 Subject: [PATCH] back to float --- imagelib/color_transfer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imagelib/color_transfer.py b/imagelib/color_transfer.py index 30e4898..ff8addb 100644 --- a/imagelib/color_transfer.py +++ b/imagelib/color_transfer.py @@ -84,7 +84,7 @@ def reinhard_color_transfer(target, source, clip=False, preserve_paper=False, ta # merge the channels together and convert back to the RGB color transfer = cv2.merge([l, a, b]) - transfer = cv2.cvtColor(transfer.astype("uint8"), cv2.COLOR_LAB2BGR) + transfer = cv2.cvtColor(transfer.astype("uint8"), cv2.COLOR_LAB2BGR).astype("float32") np.clip(transfer, 0, 1, out=transfer) # return the color transferred image