如何用概率霍夫变换检测并绘制特定角度范围的直线（Python求助）

尝试使用OpenCV的概率霍夫变换检测特定角度范围的直线，当前代码能检测到直线，但无法显示指定角度范围的结果，以下是问题分析和修正方案：

原代码的核心问题 #

1. 弧度与角度混淆：math.atan()返回的是弧度值（范围-π/2 ~ π/2），但代码中用95、165这类角度值做判断，数值范围完全不匹配，导致条件永远不成立。
2. 垂直直线处理缺失：当直线垂直时x2-x1=0，计算斜率会触发除零错误。
3. 变量未初始化风险：m仅在满足角度条件时被赋值，若没有符合条件的直线，cv2.imshow会因m未定义报错。
4. 角度范围逻辑不完整：atan无法区分方向相反的直线（比如斜率为1和-1的直线，atan返回45°和-45°），需要统一转换为0-180°的角度范围。

修正后的代码 #

import cv2
import numpy as np
import math

def gtob(image):
    cdst = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)
    cdstP = np.copy(cdst)
    return cdstP

# 假设segmentation和canny_with_threshold是已实现的函数
x = segmentation(img)
canny = canny_with_threshold(x)
z = gtob(canny)
theta = np.pi / 180
linesP = cv2.HoughLinesP(canny, 1, theta, 50, None, 150, 10)

if linesP is not None:
    for i in range(len(linesP)):
        l = linesP[i][0]
        x1, y1 = l[0], l[1]
        x2, y2 = l[2], l[3]
        
        # 计算直线角度（转换为0-180°的角度值）
        dx = x2 - x1
        dy = y2 - y1
        if dx == 0:
            # 垂直直线，角度为90°
            angle_deg = 90.0
        else:
            # 计算弧度后转换为角度，再调整到0-180°范围
            angle_rad = math.atan2(dy, dx)
            angle_deg = math.degrees(angle_rad)
            if angle_deg < 0:
                angle_deg += 180
        
        # 判断是否在目标角度范围（95°~165°）
        if 95 < angle_deg < 165:
            cv2.line(z, (x1, y1), (x2, y2), (0, 0, 255), 3, cv2.LINE_AA)

# 直接显示绘制后的z，避免m未定义的问题
cv2.imshow('ses', z)
cv2.waitKey(0)
cv2.destroyAllWindows()

关键修正点说明 #

• 使用math.atan2(dy, dx)替代atan(slope)：atan2可以根据dx和dy的正负直接计算出正确的弧度值，覆盖所有方向的直线，避免斜率无法区分方向的问题。
• 弧度转角度：通过math.degrees()将弧度转换为角度值，调整到0-180°范围，符合视觉上的直线角度认知。
• 处理垂直直线：单独判断dx=0的情况，直接赋值90°，避免除零错误。
• 统一绘制到z：始终在初始的彩色图z上绘制符合条件的直线，最后显示z，避免变量未定义的报错。

内容的提问来源于stack exchange，提问作者Rune619