Added condition to convert image to rgb only when it is rgb (ocampor#43)

Devparihar5 · Feb 12, 2022 · a0435e2 · a0435e2
1 parent d548cde
commit a0435e2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/imquality/brisque.py b/imquality/brisque.py
@@ -42,7 +42,10 @@ def __init__(
         sigma: float = 7 / 6,
     ):
         self.image = pil2ndarray(image)
-        self.image = skimage.color.rgb2gray(self.image)
+
+        if self.image.shape[-1] == 3:
+            self.image = skimage.color.rgb2gray(self.image)
+
         self.kernel_size = kernel_size
         self.sigma = sigma
         self.kernel = gaussian_kernel2d(kernel_size, sigma)