Pulled out configs. Added main.

colorbynumber/config.py

@@ -0,0 +1,11 @@
+default_config = {
+    "denoise": True,
+    "denoise_h": 100,
+    "border_padding": 2,
+    "area_perc_threshold": 0.05,
+    "arc_length_area_ratio_threshold": 1,
+    "border_color": (0, 0, 0),
+    "font_size": 0.5,
+    "font_color": (0, 0, 0),
+    "font_thickness": 2,
+}

colorbynumber/gen_islands.py

@@ -2,6 +2,9 @@ import cv2 as cv
 import numpy as np
 from polylabel import polylabel
 
+from .config import default_config
+
+
 class GenerateIslands:
     def __init__(self, indices_color_choices,):
         self.indices_color_choices = indices_color_choices
@@ -144,8 +147,11 @@ class GenerateIslands:
             self.island_borders[color_index].append((color_index, contour_border_coords))
 
     
-    def get_islands(self, border_padding=2, area_perc_threshold=0.05, 
-                    arc_length_area_ratio_threshold=0.1):
+    def get_islands(self, config = default_config):
+        border_padding = config["border_padding"]
+        area_perc_threshold = config["area_perc_threshold"]
+        arc_length_area_ratio_threshold = config["arc_length_area_ratio_threshold"]
+
         for color_index in np.unique(self.indices_color_choices):
             self._get_islands_for_one_color(
                 color_index = color_index, 

colorbynumber/main.py

@@ -0,0 +1,39 @@
+import cv2 as cv
+
+from .config import default_config
+from .simplify_image import simplify_image
+from .gen_islands import GenerateIslands
+from .numbered_islands import create_numbered_islands
+
+class ColorByNumber:
+    def __init__(self, image_path, color_list, config = default_config):
+        self.image_path = image_path
+        self.color_list = color_list
+        self.config = config
+
+    def create_color_by_number(self):
+        image = cv.imread(self.image_path)
+        image = cv.cvtColor(image, cv.COLOR_BGR2RGB)
+        self.image = image
+
+        simplified_image, indices_color_choices = simplify_image(
+            image=self.image, 
+            color_list=self.color_list,
+            config=self.config
+            )
+        self.simplified_image = simplified_image
+
+        generate_islands_obj = GenerateIslands(indices_color_choices)
+        island_borders_list, centroid_coords_list = generate_islands_obj.get_islands(config=self.config)
+        self.generate_islands_obj = generate_islands_obj
+        self.island_borders_list = island_borders_list
+
+        numbered_islands = create_numbered_islands(
+            islands = self.island_borders_list, 
+            image_shape = self.image.shape,
+            centroid_coords_list = centroid_coords_list,
+            config = self.config
+            )
+        self.numbered_islands = numbered_islands
+
+        return self.numbered_islands

colorbynumber/numbered_islands.py

@@ -1,6 +1,8 @@
 import cv2
 import numpy as np
 
+from .config import default_config
+
 def _get_centroid(coordinates):
     rows, cols = coordinates
     if len(rows) == 0 or len(cols) == 0:
@@ -8,7 +10,7 @@ def _get_centroid(coordinates):
     return (int(np.mean(rows)), int(np.mean(cols)))
 
 
-def _add_text_to_image(image, text, position, font_size=0.5, font_color=(0, 0, 0)):
+def _add_text_to_image(image, text, position, font_size, font_color, font_thickness):
     """Add text to an image.
     
     Args:
@@ -21,14 +23,12 @@ def _add_text_to_image(image, text, position, font_size=0.5, font_color=(0, 0, 0
         np.array: A new image with the text added.
     """
     font = cv2.FONT_HERSHEY_SIMPLEX
-    font_scale = font_size
-    font_thickness = 2
     return cv2.putText(
         image, 
         text, 
         position, 
         font, 
-        font_scale, 
+        font_size, 
         font_color, 
         font_thickness, 
         cv2.LINE_AA
@@ -37,8 +37,8 @@ def _add_text_to_image(image, text, position, font_size=0.5, font_color=(0, 0, 0
 
 def create_numbered_islands(islands, image_shape, 
                             centroid_coords_list = None,
-                            border_color=[0, 0, 0], 
-                            padding=2, show_numbers=True, binary = False):
+                            config = default_config, 
+                            show_numbers=True, binary = False):
     """Create a new image with the islands numbered.
     
     Args:
@@ -48,6 +48,13 @@ def create_numbered_islands(islands, image_shape,
     Returns:
         np.array: A new image with the islands numbered.
     """
+
+    padding = config["border_padding"]
+    border_color = config["border_color"]
+    font_size = config["font_size"]
+    font_color = config["font_color"]
+    font_thickness = config["font_thickness"]
+
     # Create an all white image
     width, height, channels = image_shape
     numbered_islands = np.ones((width + padding*2, height + padding*2, channels), 
@@ -64,9 +71,12 @@ def create_numbered_islands(islands, image_shape,
                 centroid = _get_centroid(island_coordinates)
             if not np.isnan(centroid).any():
                 numbered_islands = _add_text_to_image(
-                    numbered_islands, 
-                    str(color_id), 
-                    centroid
+                    image=numbered_islands, 
+                    text=str(color_id), 
+                    position=centroid,
+                    font_size=font_size,
+                    font_color=font_color,
+                    font_thickness=font_thickness
                 )
     if binary:
         # Convert numbered_islands to binary using openCV

colorbynumber/simplify_image.py

@@ -1,6 +1,7 @@
 import cv2 as cv
 import numpy as np
 
+from .config import default_config
 
 def _choose_closest_colors(image, color_list):
     """
@@ -40,8 +41,7 @@ def _denoise_image(image, h):
 
 def simplify_image(image, 
                    color_list, 
-                   denoise = True,
-                   denoise_h = 100
+                   config = default_config,
                    ):
     """
     Converts all colors in an image to the closest color in the color list.
@@ -54,6 +54,8 @@ def simplify_image(image,
     Returns:
       A copy of the image with all colors replaced with the closest color in the list.
     """
+    denoise = config["denoise"]
+    denoise_h = config["denoise_h"]
     
     simplified_image, indices_color_choices = _choose_closest_colors(image, color_list)
     if denoise:

colorbynumber/utils.py

@@ -20,3 +20,5 @@ def draw_contours(contour, image_width, image_height):
         cv.drawContours(contours_image, [contour], 0, (0,255,0), 4)
     show_image(contours_image, cmap = 'gray')
 
+def save_image(image, filename):
+    cv.imwrite(filename, image)