Merge pull request #5 from ricdodds/fix/color-correction

Color correction
nadeemlab · Oct 6, 2021 · aa6940e · aa6940e
2 parents c4bf68f + 8334d72
commit aa6940e
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/deepliif/util/util.py b/deepliif/util/util.py
@@ -44,14 +44,8 @@ def tensor2im(input_image, imtype=np.uint8):
     return image_numpy.astype(imtype)
 
 
-def cv_to_pil(img):
-    return Image.fromarray(
-        cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-    ).convert('RGB')
-
-
 def tensor_to_pil(t):
-    return cv_to_pil(tensor2im(t))
+    return Image.fromarray(tensor2im(t))
 
 
 def diagnose_network(net, name='network'):