Python LSD直线检测算法

import cv2

# 读取输入图像
image = cv2.imread('images/Snap_003.jpg')

# 转换图像为灰度
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# 应用高斯模糊
gaussian_blur_image = cv2.GaussianBlur(gray, (5, 5), 0)
# 创建LSD行段检测器对象
lsd = cv2.createLineSegmentDetector(0)

# 检测直线
lines, width, prec, nfa = lsd.detect(gaussian_blur_image)

# 在图像上绘制检测到的直线
drawn_img = lsd.drawSegments(image, lines)

# 显示绘制了直线的图像
cv2.imshow('LSD Lines', drawn_img)
cv2.waitKey(0)
cv2.destroyAllWindows()