正交骨架线法python代码

正交骨架线法（Orthogonal Skeleton Extraction）是一种图像处理算法，可以用于提取图像中的骨架线。它的原理是通过对图像进行分析，找到具有代表性的像素点，然后将这些点连接起来，形成骨架线。

以下是一个简单的Python代码实现：

import cv2
import numpy as np

# 读取图像
img = cv2.imread('test.jpg', cv2.IMREAD_GRAYSCALE)

# 二值化处理
_, img = cv2.threshold(img, 127, 255, cv2.THRESH_BINARY)

# 计算距离变换
dist_transform = cv2.distanceTransform(img, cv2.DIST_L2, 3)

# 极大值抑制
local_maxima = np.zeros(dist_transform.shape, dtype=np.uint8)
local_maxima[dist_transform == cv2.dilate(dist_transform, np.ones((3, 3)), iterations=1)] = 255

# 提取骨架线
skel = np.zeros(dist_transform.shape, dtype=np.uint8)
while cv2.countNonZero(local_maxima) > 0:
    eroded = cv2.erode(local_maxima, np.ones((3, 3)))
    temp = cv2.dilate(eroded, np.ones((3, 3))) - eroded
    skel |= temp
    local_maxima &= ~temp

# 显示结果
cv2.imshow('Skeleton', skel)
cv2.waitKey(0)
cv2.destroyAllWindows()

这段代码的主要流程如下：

1. 读取图像并进行二值化处理。
2. 计算距离变换，找到距离图像中每个像素点最近的背景像素点的距离。
3. 对距离图像进行极大值抑制，得到代表骨架线的像素点。
4. 不断进行局部极大值抑制和腐蚀操作，直到无法再得到新的骨架线像素点。
5. 显示提取出的骨架线图像。