基于OpenCV的水平线检测及间距测量相关技术问题咨询

用户问题背景 #

我是OpenCV新手，目前尝试使用OpenCV的HoughLinesP函数检测图像中的水平线，以计算x与y的比例，检测结果见结果图。

• 原图像：原图像
• 结果图像：结果图像

技术问题 #

1. 如何忽略垂直线，仅检测水平线？
2. HoughLinesP检测出的边缘是多段线，有没有更好的方法将其合并为单条线？
3. 下一步应如何计算x与y的比例？

当前实现代码 #

import cv2
import numpy as np

# read image
img = cv2.imread('testtube.png')

# convert to grayscale
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

# apply canny edge detection
edges = cv2.Canny(gray, 90, 100,apertureSize=3)

# get hough lines
result = img.copy()

lines = cv2.HoughLinesP(edges, rho=1,theta=np.pi/2, threshold=30,lines=np.array([]), minLineLength=30,maxLineGap=20)
a,b,c = lines.shape
for i in range(a):
    cv2.line(result, (lines[i][0][0], lines[i][0][1]), (lines[i][0][2], lines[i][0][3]), (0, 0, 255), 2, cv2.LINE_AA)

cv2.imshow("edges", edges)
cv2.imshow("result", result)
cv2.waitKey(0)
cv2.destroyAllWindows()

问题解答 #

1. 仅检测水平线的方法 #

你当前设置的theta=np.pi/2是检测垂直线的参数，水平线对应的theta应该是0（或np.pi，因为Hough变换的theta是从x轴正方向逆时针旋转的角度）。另外还可以通过过滤线段斜率进一步筛选：

• 水平线的y坐标变化极小，计算线段的dy = abs(y2 - y1)，如果dy < 5（阈值可根据图像调整），则判定为水平线，否则过滤掉。

示例代码片段：

# 调整theta为0，优先检测水平线
lines = cv2.HoughLinesP(edges, rho=1, theta=0, threshold=30, minLineLength=50, maxLineGap=10)

# 过滤出纯水平线
horizontal_lines = []
for line in lines:
    x1, y1, x2, y2 = line[0]
    if abs(y2 - y1) < 5:
        horizontal_lines.append(line)

2. 合并多段水平线的方法 #

两种实用思路：

• 调参优化：增大minLineLength（比如从30改为50）、减小maxLineGap（比如从20改为10），让HoughLinesP直接检测出更长的连续线段。
• 聚类合并：因为是水平线，y坐标相近的线段属于同一条线。将所有水平线按y坐标分组（y值差在5以内归为一组），每组取最小x和最大x作为端点，合并成完整线段。

合并逻辑示例：

if horizontal_lines:
    # 按y坐标排序线段
    horizontal_lines.sort(key=lambda l: l[0][1])
    merged_lines = []
    current_line = horizontal_lines[0][0]
    
    for line in horizontal_lines[1:]:
        x1, y1, x2, y2 = line[0]
        # 若当前线段与合并线y值接近，则扩展合并线的x范围
        if abs(y1 - current_line[1]) < 5:
            current_line[0] = min(current_line[0], x1)
            current_line[2] = max(current_line[2], x2)
        else:
            merged_lines.append(current_line)
            current_line = line[0]
    merged_lines.append(current_line)
    
    # 绘制合并后的线
    for line in merged_lines:
        cv2.line(result, (line[0], line[1]), (line[2], line[3]), (0, 255, 0), 2, cv2.LINE_AA)

3. 计算x与y比例的步骤 #

以试管类物体为例：

1. 提取关键线段：找到两条核心水平线（比如试管顶部和底部边缘），记录它们的x方向长度（像素）和y方向垂直距离（像素差）。
2. 结合真实尺寸计算：若知道物体真实尺寸（如试管真实高度H、真实宽度W）：
   • 像素宽度：pixel_width = 顶部线段x2 - 顶部线段x1
   • 像素高度：pixel_height = 底部线段y - 顶部线段y
   • x与y的比例 = (pixel_width / W) : (pixel_height / H)，简化为 (pixel_width * H) : (pixel_height * W)
3. 无真实尺寸的情况：用图像中已知长度的参考物（如标尺）算出像素与真实长度的转换系数，再换算目标物体的x/y比例。

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