multimodalart HF staff commited on
Commit
fb8832c
1 Parent(s): e889613

Update sketch_helper.py

Browse files
Files changed (1) hide show
  1. sketch_helper.py +1 -0
sketch_helper.py CHANGED
@@ -3,6 +3,7 @@ import cv2
3
  from PIL import Image
4
  from skimage.color import rgb2lab
5
  from skimage.color import lab2rgb
 
6
 
7
  def color_quantization(image, n_colors):
8
  # Convert image to LAB color space
 
3
  from PIL import Image
4
  from skimage.color import rgb2lab
5
  from skimage.color import lab2rgb
6
+ from sklearn.cluster import KMeans
7
 
8
  def color_quantization(image, n_colors):
9
  # Convert image to LAB color space