StephaneBah commited on
Commit
35f5b1e
1 Parent(s): 45eafb1

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -1,7 +1,7 @@
1
  import numpy as np
2
  import cv2
3
 
4
- def kmeans(image, k=4, alpha=0.5):
5
  image = np.array(image)
6
  pixel_vals = image.reshape((-1,3))
7
  pixel_vals = np.float32(pixel_vals)
 
1
  import numpy as np
2
  import cv2
3
 
4
+ def kmeans(image, k=3, alpha=0.5):
5
  image = np.array(image)
6
  pixel_vals = image.reshape((-1,3))
7
  pixel_vals = np.float32(pixel_vals)