wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-20 12:40:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix initialization in km_segmentation, prettier examples

Browse Source
This commit is contained in:
Andreas Mueller
2012-08-03 11:43:03 +01:00
parent d9a22d867b
commit 026b6b1df0
7 changed files with 46 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/examples/plot_km_segmentation.py
+5 -20
View File
@@ -6,31 +6,16 @@ import matplotlib.pyplot as plt
import numpy as np
from skimage.data import lena
from skimage.segmentation import km_segmentation
from skimage.segmentation import km_segmentation, visualize_boundaries
from skimage.util import img_as_float
from skimage.color import rgb2lab
img = img_as_float(lena()).copy("C")
segments = km_segmentation(img, ratio=2.0, n_segments=200)
segments = km_segmentation(rgb2lab(img), ratio=10.0, n_segments=1000)
print("number of segments: %d" % len(np.unique(segments)))
plt.subplot(131, title="original")
plt.imshow(img, interpolation='nearest')
boundaries_mine = visualize_boundaries(img, segments)
plt.imshow(boundaries_mine)
plt.axis("off")
plt.subplot(132, title="superpixels")
# shuffle the labels for better visualization
plt.imshow(segments, interpolation='nearest', cmap=plt.cm.prism)
plt.axis("off")
plt.subplot(133, title="mean color")
colors = [np.bincount(segments.ravel(), img[:, :, c].ravel()) for c in
xrange(img.shape[2])]
counts = np.bincount(segments.ravel())
colors = np.vstack(colors) / counts
plt.imshow(colors.T[segments], interpolation='nearest')
plt.axis("off")
plt.subplots_adjust(wspace=0.02, hspace=0.02, top=0.9,
bottom=0.02, left=0.02, right=0.98)
plt.show()