Implemented new method to detect color images

kcc/image.py

@@ -442,28 +442,24 @@ class ComicPage:
                 self.fill = 'white'
 
     def isImageColor(self):
-        v = ImageStat.Stat(self.image).var
+        img = self.image.copy()
-        isMonochromatic = reduce(lambda x, y: x and y < 0.005, v, True)
+        bands = img.getbands()
-        if isMonochromatic:
+        if bands == ('R', 'G', 'B') or bands == ('R', 'G', 'B', 'A'):
-            # Monochromatic
+            thumb = img.resize((40, 40))
+            SSE, bias = 0, [0, 0, 0]
+            bias = ImageStat.Stat(thumb).mean[:3]
+            bias = [b - sum(bias) / 3 for b in bias]
+            for pixel in thumb.getdata():
+                mu = sum(pixel) / 3
+                SSE += sum((pixel[i] - mu - bias[i]) * (pixel[i] - mu - bias[i]) for i in [0, 1, 2])
+            MSE = float(SSE) / (40 * 40)
+            if MSE <= 22:
                 return False
             else:
-            if len(v) == 3:
-                maxmin = abs(max(v) - min(v))
-                if maxmin > 1000:
-                    # Color
                 return True
-                elif maxmin > 100:
+        elif len(bands) == 1:
-                    # Probably color
-                    return True
-                else:
-                    # Grayscale
-                    return False
-            elif len(v) == 1:
-                # Black and white
             return False
         else:
-                # Detection failed
             return False