Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -102,7 +102,7 @@ def dominant_color_descriptor(image, k=3):
|
|
102 |
pixels = image.reshape(-1, 3)
|
103 |
|
104 |
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 100, 0.2)
|
105 |
-
flags = cv2.KMEANS_RANDOM_CENTERS
|
106 |
|
107 |
try:
|
108 |
_, labels, centers = cv2.kmeans(pixels.astype(np.float32), k, None, criteria, 10, flags)
|
@@ -187,7 +187,7 @@ def preprocess_image(image):
|
|
187 |
# # Convert to RGB if needed
|
188 |
# if image.mode != 'RGB':
|
189 |
# image = image.convert('RGB')
|
190 |
-
# Convert to numpy array
|
191 |
img_array = np.array(image)
|
192 |
|
193 |
# # Convert to RGB if needed
|
|
|
102 |
pixels = image.reshape(-1, 3)
|
103 |
|
104 |
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 100, 0.2)
|
105 |
+
flags = cv2.KMEANS_RANDOM_CENTERS
|
106 |
|
107 |
try:
|
108 |
_, labels, centers = cv2.kmeans(pixels.astype(np.float32), k, None, criteria, 10, flags)
|
|
|
187 |
# # Convert to RGB if needed
|
188 |
# if image.mode != 'RGB':
|
189 |
# image = image.convert('RGB')
|
190 |
+
# Convert to numpy array
|
191 |
img_array = np.array(image)
|
192 |
|
193 |
# # Convert to RGB if needed
|