SonFox2920 commited on
Commit
972c8da
·
verified ·
1 Parent(s): 398bb14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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