python的项目骨架_用Python对图像进行骨架化

import scipy.ndimage.morphology as m

import numpy as np

import cv2

def skeletonize(img):

h1 = np.array([[0, 0, 0],[0, 1, 0],[1, 1, 1]])

m1 = np.array([[1, 1, 1],[0, 0, 0],[0, 0, 0]])

h2 = np.array([[0, 0, 0],[1, 1, 0],[0, 1, 0]])

m2 = np.array([[0, 1, 1],[0, 0, 1],[0, 0, 0]])

hit_list = []

miss_list = []

for k in range(4):

hit_list.append(np.rot90(h1, k))

hit_list.append(np.rot90(h2, k))

miss_list.append(np.rot90(m1, k))

miss_list.append(np.rot90(m2, k))

img = img.copy()

while True:

last = img

for hit, miss in zip(hit_list, miss_list):

hm = m.binary_hit_or_miss(img, hit, miss)

img = np.logical_and(img, np.logical_not(hm))

if np.all(img == last):

break

return img

img = cv2.imread("e_5.