Added apply_kmeans config

colorbynumber/config.py

@@ -6,6 +6,10 @@ default_config = {
     # Options: "before_simplify", "after_simplify"
     "denoise_order": "before_simplify",
 
+    # If True, the image will be simplified using kmeans clustering.
+    # And then the colors will be matched to closest color in the palette.
+    "apply_kmeans": True,
+
     # Type of denoising to be used.
     # Options: "fastNlMeansDenoisingColored", "gaussianBlur", "blur"
     "denoise_type": "gaussianBlur",

colorbynumber/simplify_image.py

@@ -102,6 +102,8 @@ def simplify_image(image,
         simplified_image, indices_color_choices, color_list = _kmeans_simplify_image(image, num_colors)
 
     else:
+        if config["apply_kmeans"]:
+            image, indices_color_choices, color_list_kmeans = _kmeans_simplify_image(image, len(color_list))
         simplified_image, indices_color_choices = _choose_closest_colors(image, color_list)