OpenCV 霍夫线检测练习直线绘制代码调整

概述

黑马程序员人工智能教程_10小时学会图像处理OpenCV入门教程中，3.6霍夫线检测代码，关于直线绘制的部分，没有看懂，这里，根据自己的理解，对直线绘制的代码进行了实现。

原理与实现

对于笛卡尔坐标系下y = ax + b，转换成极坐标系下有rho = x * cos(theta) + y * sin(theta),
两边除以sin(theta)得到下式:y = - cos(theta) / sin(theta) * x + rho / sin(theta)
当sin(theta) ==0时，有x = rho / cos(theta)
当sin(theta) !=0时，有a = - cos(theta) / sin(theta) ; b = rho / sin(theta)
由a, b可以得到直角坐标系下直线方程，可以得到两个点，即可画出直线。
在图像上绘制直线的代码如下所示，最后的效果，与老师视频中的效果一致。代码如下所示

print('lines.shape =', lines.shape)
h, w = img.shape[:2]
for line in lines:
    rho, theta = line[0]
    if math.sin(theta) == 0:
        x = int(rho / math.cos(theta))
        cv.line(img, (x, 0), (x, h - 1), (0, 255, 0))
    else:            
        a = - math.cos(theta) / math.sin(theta)
        b = rho / math.sin(theta)    
        # y = a * x + b，计算直线中两个点
        x1 = 0
        y1 = int(b)
        x2 = w - 1
        y2 = int(a * x1 + b)
        cv.line(img, (x1, y1), (x2, y2), (0, 255, 0))

参考文献

https://zhuanlan.zhihu.com/p/... (hough变换原理以及实现（转载） - 知乎)