Added image downsampling

colorbynumber/main.py

@@ -2,7 +2,7 @@ import cv2 as cv
 import numpy as np
 
 from .config import default_config
-from .simplify_image import simplify_image
+from .simplify_image import simplify_image, downsample_image
 from .gen_islands import GenerateIslands
 from .numbered_islands import create_islands, add_numbers_to_image
 
@@ -26,6 +26,7 @@ class ColorByNumber:
 
         image = cv.imread(self.image_path)
         image = cv.cvtColor(image, cv.COLOR_BGR2RGB)
+        image = downsample_image(image)
         self.image = image
 
     def create_color_by_number(self):

colorbynumber/simplify_image.py

@@ -73,6 +73,22 @@ def _denoise_image(image, h, denoise_type, blur_size = None):
 
     return denoised_image
 
+def downsample_image(image):
+    """
+    Downsample the image so the max dimension is 1000 pixels.
+    """
+    max_dim = 1000
+    width, height = image.shape[:2]
+    if width > height:
+        new_width = max_dim
+        new_height = int(height * (new_width / width))
+    else:
+        new_height = max_dim
+        new_width = int(width * (new_height / height))
+    
+    image = cv.resize(image, (new_height, new_width), interpolation = cv.INTER_AREA)
+    return image
+
 def simplify_image(image, 
                    color_list = None,
                    num_colors = None,