在 opencv 中使用 houghlines 只打印一行

In opencv using houghlines prints only one line

我开始学习一些关于 opencv 的教程并研究 houghlines,并注意到我给它的任何图像都只会 return 一行!

我用的是opencv 4.2.0,我的代码是:

import cv2
import numpy as np

image =cv2.imread("sudoku.jpg")
gray=cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
edges=cv2.Canny(gray, 100, 170,apertureSize=3)
cv2.imshow(" lines",edges)
cv2.waitKey()
cv2.destroyAllWindows()

lines=cv2.HoughLines(edges, 1, np.pi/180, 240)

for rho,theta in lines[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))
    cv2.line(image,(x1,y1),(x2,y2),(255,0,0),2)

cv2.imshow("hough lines",image)
cv2.waitKey()
cv2.destroyAllWindows()

实际上,数据在 lines 变量中的存储方式已在较新版本的 OpenCV 中更新,因此您正面临此问题。

使用下面的嵌套 for 循环代替 for 循环在图像上绘制所有线条:

for line in lines:
    for rho,theta in line:
        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))
        cv2.line(image,(x1,y1),(x2,y2),(255,0,0),2)

要查看数据的存储方式,您可以打印 lines 变量。