霍夫变换多线检测场景仅输出单条线条的问题排查与修正咨询

• 线条绘制逻辑错误：代码中仅遍历lines[0]，只会绘制霍夫变换检测到的第一条线，这是只输出单条线的核心原因。
• 缺少空值判断：未对霍夫变换返回的lines做非空校验，当无符合条件的线条时直接访问lines[0]会触发报错，是部分图片无输出的直接原因。
• 预处理逻辑冗余且效果差：通过Sobel梯度幅值转换生成灰度图的流程反而会削弱路牌边缘特征，且未启用预设的颜色过滤逻辑，背景干扰过多导致有效边缘被淹没。
• 参数设置过于严格：霍夫变换的投票阈值设置为200，路牌边缘长度不足无法达到投票数要求，大量有效线条被过滤。

• 修正线条遍历逻辑：确认lines不为空后，遍历所有检测到的线条绘制。
• 优化预处理流程：将原图转为HSV颜色空间，过滤蓝色路牌区域生成掩码，过滤背景干扰后再做边缘检测，去掉冗余的Sobel梯度转换步骤，直接对掩码后的灰度图做Canny边缘检测。
• 调整算法参数：
  • 适当降低Canny边缘检测的高低阈值，保留更多弱边缘
  • 降低霍夫变换的投票阈值到50~100区间，可根据实际效果微调，同时可缩小角度步长到np.pi/180提升角度检测精度
• 增加空值校验：检测lines不为空再执行绘制逻辑，避免程序运行异常。

import numpy as np
import cv2 as cv
import os

images = []
# 蓝色路牌HSV阈值范围
lower_blue = np.array([90, 70, 70])
upper_blue = np.array([130, 255, 255])

# 导入测试图片
for i in os.listdir("Directional Signage/Train"):
    img = cv.imread(os.path.join("Directional Signage/Train", i))
    if img is None:
        print("Couldn't read this image" + str(i))
    else:
        images.append(img)

for i in range(len(images)):
    img = images[i].copy()
    # 颜色过滤提取路牌区域
    hsv = cv.cvtColor(img, cv.COLOR_BGR2HSV)
    mask = cv.inRange(hsv, lower_blue, upper_blue)
    # 掩码后转灰度做边缘检测
    gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
    masked_gray = cv.bitwise_and(gray, gray, mask=mask)
    edges = cv.Canny(masked_gray, 30, 90, apertureSize=3)
    # 调整霍夫变换参数
    lines = cv.HoughLines(edges, 1, np.pi/180, 80)
    # 非空判断后遍历所有线条绘制
    if lines is not None:
        for line in lines:
            rho, theta = line[0]
            a = np.cos(theta)
            b = np.sin(theta)
            x0 = a*rho
            y0 = b*rho
            x1 = int(x0 + 1000*(-b))
            y1 = int(y0 + 1000*(a))
            x2 = int(x0 - 1000*(-b))
            y2 = int(y0 - 1000*(a))
            cv.line(img, (x1,y1), (x2,y2), (0,0,255), 2)
    images[i] = img

for i in range(len(images)):
    cv.imshow("blobby " + str(i), images[i])

l = cv.waitKey(0)

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