0

所以我想做的是使用霍夫变换在图像上找到线条，然后将线条绘制到图像中。

但我收到一条错误消息：

Traceback (most recent call last): 
    File "houghLines.py", line 31, in <module>
    main() File "houghLines.py", line 13, in main 
    a = hough(image)
    File "houghLines.py", line 29, in hough
    (0,0,255))
    TypeError: only length-1 arrays can be converted to Python scalars

这是我的代码：

import cv2
import os
import math
def main():

    directory = os.path.dirname(__file__)
    picLoc = os.path.join(directory, "../video-image/1m50s.png")

    image = cv2.imread(picLoc)
    print "sending image to houghLines.py"

    a = hough(image)
    cv2.imshow("", a)
    cv2.waitKey(0)
    cv2.destroyAllChildren()

def hough(image):
    canny = cv2.Canny(image, 50, 200)
    color_image = cv2.cvtColor(canny, cv2.COLOR_GRAY2BGR)
    houghLines = cv2.HoughLinesP(canny, 1, math.pi/180, 50)
    for x in range(len(houghLines)):
        print x
        pt1 = (houghLines[x][0], houghLines[x][1])
        pt2 = (houghLines[x][2], houghLines[x][3])
        cv2.line(color_image, pt1, pt2, (0,0,255), 3)
    return color_image

main()
4

1 回答 1

1

出现此问题的原因是cv2.line()期望pt1和pt2是元组，每个元组只包含一个元素x和y坐标。如果您print pt1在您的示例中，您会很快发现情况并非如此。

cv2.HoughLinesP()返回一个只有一个元素的 numpy 数组。此元素包含四个点的列表，这些点是您的线的起点和终点。知道了这一点，正确的实现如下：

    for line in houghLines[0]:
        pt1 = tuple(line[:2])
        print pt1
        pt2 = tuple(line[-2:])
        cv2.line(color_image, pt1, pt2, (0,0,255), 3)
于 2013-07-10T16:38:22.677 回答