我正在尝试用https://homepages.inf.ed.ac.uk/rbf/HIPR2/thin.htm中描述的细化/命中或未命中方法实现骨架化算法
我想从下面的图像中提取骨架,我认为这是一个很好的测试图像:
https://github.com/TiagoB936/Image-Processing-and-Computer-Vision/blob/master/Skeletonization/x.png
问题是我的代码无法停止运行,我没有发现问题所在
我的代码执行以下步骤:
创建2个主内核:
[[0,0,0]
[-1,1,-1]
[1,1,1]]
以及
[-1,0,0]
[1,1,0]
[-1,1,1]]
创建另外6个内核,每个内核与主内核旋转90度
下面是一些测试代码:
import cv2 as cv
import numpy as np
input_image = cv.imread('x.png',0)
_,input_image = cv.threshold(input_image,60,255,cv.THRESH_BINARY)
kernel1 = np.array((
[0, 0, 0],
[-1, 1, -1],
[1, 1, 1]), dtype="int")
kernel2 = np.array((
[-1, 0, 0],
[1, 1, 0],
[-1, 1, -1]), dtype="int")
kernel3 = np.rot90(kernel1)
kernel4 = np.rot90(np.rot90(kernel1))
kernel5 = np.rot90(np.rot90(np.rot90(kernel1)))
kernel6 = np.rot90(kernel2)
kernel7 = np.rot90(np.rot90(kernel2))
kernel8 = np.rot90(np.rot90(np.rot90(kernel2)))
output_image_1 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel1)
output_image_2 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel2)
output_image_3 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel3)
output_image_4 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel4)
output_image_5 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel5)
output_image_6 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel6)
output_image_7 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel7)
output_image_8 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel8)
hit_miss = output_image_1 + output_image_2 + output_image_3 + output_image_4 + output_image_5 + output_image_6 + output_image_7 + output_image_8
output_image = input_image - hit_miss
while not np.array_equal(output_image, input_image):
input_image = output_image
output_image_1 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel1)
output_image_2 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel2)
output_image_3 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel3)
output_image_4 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel4)
output_image_5 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel5)
output_image_6 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel6)
output_image_7 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel7)
output_image_8 = cv.morphologyEx(input_image, cv.MORPH_HITMISS, kernel8)
hit_miss = output_image_1 + output_image_2 + output_image_3 + output_image_4 + output_image_5 + output_image_6 + output_image_7 + output_image_8
output_image = input_image - hit_miss
cv.imshow("hit_miss", hit_miss)
cv.imshow("out", output_image)
cv.waitKey(0)
cv.destroyAllWindows()
我本来希望从我的输入中得到一个瘦的X图像,但是代码无法停止运行。我错过什么了吗
您希望在应用所有内核之后使用逻辑或,而不是仅仅添加它们
相关问题 更多 >
编程相关推荐