Spaces:

weizmannscience
/

multidiffusion-region-based

Sleeping

multimodalart HF staff commited on Feb 26, 2023

Commit

f07954b

•

1 Parent(s): 7b0ac85

Update sketch_helper.py

Files changed (1) hide show

sketch_helper.py CHANGED Viewed

@@ -35,7 +35,7 @@ def color_quantization(image, n_colors=8, rounds=1):
             count += 1
     compactness, labels, centers = cv2.kmeans(samples,
-            clusters,
             None,
             (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10000, 0.0001),
             rounds,

             count += 1
     compactness, labels, centers = cv2.kmeans(samples,
+            n_colors,
             None,
             (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10000, 0.0001),
             rounds,