mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-28 04:55:16 +08:00
Neil Yager
53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
"""
|
|
===========
|
|
Skeletonize
|
|
===========
|
|
|
|
An example of thinning a binary image using skeletonize.
|
|
"""
|
|
from scikits.image.morphology import skeletonize
|
|
from scikits.image.draw import draw
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
# an empty image
|
|
image = np.zeros((400, 400))
|
|
|
|
# foreground object 1
|
|
image[10:-10, 10:100] = 1
|
|
image[-100:-10, 10:-10] = 1
|
|
image[10:-10, -100:-10] = 1
|
|
|
|
# foreground object 2
|
|
rs, cs = draw.bresenham(250, 150, 10, 280)
|
|
for i in range(10): image[rs+i, cs] = 1
|
|
rs, cs = draw.bresenham(10, 150, 250, 280)
|
|
for i in range(20): image[rs+i, cs] = 1
|
|
|
|
# foreground object 3
|
|
ir, ic = np.indices(image.shape)
|
|
circle1 = (ic - 135)**2 + (ir - 150)**2 < 30**2
|
|
circle2 = (ic - 135)**2 + (ir - 150)**2 < 20**2
|
|
image[circle1] = 1
|
|
image[circle2] = 0
|
|
|
|
# perform skeletonization
|
|
skeleton = skeletonize.skeletonize(image)
|
|
|
|
# display results
|
|
plt.figure(figsize=(10,6))
|
|
|
|
plt.subplot(121)
|
|
plt.imshow(image, cmap=plt.cm.gray)
|
|
plt.axis('off')
|
|
plt.title('original', fontsize=20)
|
|
|
|
plt.subplot(122)
|
|
plt.imshow(skeleton, cmap=plt.cm.gray)
|
|
plt.axis('off')
|
|
plt.title('skeleton', fontsize=20)
|
|
|
|
plt.subplots_adjust(wspace=0.02, hspace=0.02, top=0.98,
|
|
bottom=0.02, left=0.02, right=0.98)
|
|
|
|
plt.show() |