diff --git a/samplelib/SampleProcessor.py b/samplelib/SampleProcessor.py index 0ab64d6..7c42633 100644 --- a/samplelib/SampleProcessor.py +++ b/samplelib/SampleProcessor.py @@ -221,11 +221,14 @@ class SampleProcessor(object): ct_sample_bgr_resized = cv2.resize( ct_sample_bgr, (resolution,resolution), cv2.INTER_LINEAR ) ct_sample_mask_resized = cv2.resize( ct_sample_mask, (resolution,resolution), cv2.INTER_LINEAR )[...,np.newaxis] - img_bgr = imagelib.reinhard_color_transfer ( np.clip( (img_bgr*255) .astype(np.uint8), 0, 255), - np.clip( (ct_sample_bgr_resized*255).astype(np.uint8), 0, 255), - source_mask=img_mask, target_mask=ct_sample_mask_resized) + img_bgr = imagelib.linear_color_transfer (img_bgr, ct_sample_bgr_resized) + img_bgr = np.clip( img_bgr, 0.0, 1.0) + + # img_bgr = imagelib.reinhard_color_transfer ( np.clip( (img_bgr*255) .astype(np.uint8), 0, 255), + # np.clip( (ct_sample_bgr_resized*255).astype(np.uint8), 0, 255) ) + # #source_mask=ct_sample_mask_resized, target_mask=img_mask) - img_bgr = np.clip( img_bgr.astype(np.float32) / 255.0, 0.0, 1.0) + # img_bgr = np.clip( img_bgr.astype(np.float32) / 255.0, 0.0, 1.0) if normalize_std_dev: img_bgr = (img_bgr - img_bgr.mean( (0,1)) ) / img_bgr.std( (0,1) )