帮我写一个python检测直线并延长的程序

以下是一个简单的Python程序，用于检测直线并延长它：

import cv2
import numpy as np

# 读取图片
img = cv2.imread('test.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 检测直线
edges = cv2.Canny(gray, 50, 150, apertureSize=3)
lines = cv2.HoughLinesP(edges, 1, np.pi/180, 100, minLineLength=100, maxLineGap=10)

# 延长直线
for line in lines:
    x1, y1, x2, y2 = line[0]
    if x1 == x2:  # 如果是垂直线
        cv2.line(img, (x1, 0), (x2, img.shape[0]), (0, 0, 255), 2)
    else:
        k = (y2 - y1) / (x2 - x1)  # 计算斜率
        b = y1 - k * x1  # 计算截距
        x1_new = int((0 - b) / k)
        x2_new = int((img.shape[0] - b) / k)
        cv2.line(img, (x1_new, 0), (x2_new, img.shape[0]), (0, 0, 255), 2)

# 显示图片
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

该程序首先读取一张图片，然后使用Canny算子和霍夫变换检测直线。接着，程序会遍历每条直线，并根据其斜率和截距计算出新的直线端点，以延长直线。最后，程序会在原图上绘制出延长后的直线，并将其显示出来。

请注意，该程序仅适用于检测水平或垂直的直线。如果您需要检测斜直线，请使用霍夫变换的标准函数cv2.HoughLines()。