Segmen Reguler Menggunakan Watershed

 Segmen Reguler Menggunakan Watershed

import numpy as np

from skimage import data, util, filters, color

from skimage.segmentation import watershed

import matplotlib.pyplot as plt

coins = data.coins()

edges = filters.sobel(coins)

grid = util.regular_grid(coins.shape, n_points=468)

seeds = np.zeros(coins.shape, dtype=int)

seeds[grid] = np.arange(seeds[grid].size).reshape(seeds[grid].shape) + 1

w0 = watershed(edges, seeds)

w1 = watershed(edges, seeds, compactness=0.01)

fig, (ax0, ax1) = plt.subplots(1, 2)

ax0.imshow(color.label2rgb(w0, coins, bg_label=-1))

ax0.set_title('Classical watershed')

ax1.imshow(color.label2rgb(w1, coins, bg_label=-1))

ax1.set_title('Compact watershed')

plt.show()

Share on :

Tweet