diff --git a/beadDetect.py b/beadDetect.py index 829bde8..49b5d43 100755 --- a/beadDetect.py +++ b/beadDetect.py @@ -214,7 +214,7 @@ Z = np.float32(plotColors) # Define criteria = ( type, max_iter, epsilon ) criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 100, 0.1) -ret,label,center=cv2.kmeans(Z,10,None,criteria,10,cv2.KMEANS_RANDOM_CENTERS) +ret,label,center=cv2.kmeans(Z,10,None,criteria,10,cv2.KMEANS_PP_CENTERS) print (center) @@ -222,7 +222,7 @@ print (center) fig = plt.figure() ax = fig.add_subplot(111, projection='3d') ax.scatter(blue, green, red, zdir='z', s=10, c=(0,0,0), lw = 0, depthshade=True) -ax.scatter(center[:,2], center[:,1], center[:,0], zdir='z', s=250, facecolors=center, lw = 2, depthshade=True) +ax.scatter(center[:,2], center[:,1], center[:,0], zdir='z', s=250, facecolors=center, lw = 1, depthshade=True) plt.show()