Update sketch_helper.py

sketch_helper.py

@@ -5,20 +5,6 @@ from skimage.color import rgb2lab
 from skimage.color import lab2rgb
 from sklearn.cluster import KMeans
 
-def color_quantization(image, n_colors):
-    # Reshape the image into a 2D array of pixels
-    pixels = image.reshape(-1, 3)
-
-    # Fit k-means clustering algorithm to the pixel data
-    kmeans = KMeans(n_clusters=n_colors, random_state=0).fit(pixels)
-
-    # Replace each pixel with its nearest cluster center
-    labels = kmeans.predict(pixels)
-    colors = kmeans.cluster_centers_.astype(np.uint8)
-    quantized_image = colors[labels].reshape(image.shape)
-
-    return quantized_image
-
     
 def get_high_freq_colors(image):
   im = image.getcolors(maxcolors=1024*1024)
@@ -30,7 +16,7 @@ def get_high_freq_colors(image):
   high_freq_colors = [c for c in sorted_colors if c[0] > max(2, mean_freq)]  # Ignore colors that occur very few times (less than 2) or less than half the average frequency
   return high_freq_colors
 
-def color_quantization_old(image, n_colors):
+def color_quantization(image, n_colors):
     # Get color histogram
     hist, _ = np.histogramdd(image.reshape(-1, 3), bins=(256, 256, 256), range=((0, 256), (0, 256), (0, 256)))
     # Get most frequent colors