python - 像素操作后的HoughCircles错误

def detect_circles():
    img = cv2.imread('img.JPG', 0)
    #img = cv2.medianBlur(img, 3)
    #cim= cv2.GaussianBlur(img, (15, 15), 0)
    cimg = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)

    circles = cv2.HoughCircles(img, cv2.cv.CV_HOUGH_GRADIENT, 1, 30,param1=100,param2=39,minRadius=25,maxRadius=70)
    circles = np.uint16(np.around(circles))
    count = 0
    for i in circles[0, :]:
        count = count + 1
        # draw the outer circle
        cv2.circle(cimg, (i[0], i[1]), i[2], (0, 255, 0), 2)
        # draw the center of the circle
        cv2.circle(cimg, (i[0], i[1]), 2, (0, 0, 255), 3)

    cv2.imwrite('circles_detected.JPG', cimg)
    print(count)

此功能旨在检测图像中的圆圈。
它工作正常，但我需要区分颜色。
所以我写了这个函数，将每个像素的G和R值设置为0。

def iterate_image():
    img = read_img('SDC10004.JPG')
    height = img.shape[0]
    width = img.shape[1]
    for i in range(height):
        for j in range(width):
            #img.itemset((i,j, 0), 100)
            img.itemset((i,j, 1), 0)
            img.itemset((i,j, 2), 0)

    write_img(img,'SDC10004_selfmade_blue.JPG')

当我尝试检测带有蓝色像素图像的圆圈时，出现以下错误消息:

Traceback (most recent call last):
File "/home/user/Documents/workspace/ImageProcessing/Main.py", line 107, in <module>
detect_circles();
File "/home/user/Documents/workspace/ImageProcessing/Main.py", line 94, in detect_circles
circles = np.uint16(np.around(circles))
File "/usr/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 2610, in around
return _wrapit(a, 'round', decimals, out)
File "/usr/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 43, in _wrapit
result = getattr(asarray(obj), method)(*args, **kwds)
AttributeError: rint

有人曾对此进行过启发吗？

最佳答案

因此，我相信我发现了问题所在。

HoughCircles并不棘手，您只需要注意它的参数化。该方法基于Canny边缘检测器，该检测器又将灰度图像阈值化。

opencv用于灰度图像的formula由下式给出:

RGB_to_Gray: 0.299.R + 0.587.G + 0.114.B

由于您要将三个RGB通道中的两个清零，因此灰度值的强度将降低。结果，您的阈值避免了任何圆的检测，并且 cv2.HoughCircles 返回无。因此，由于您不测试圈子类型是否为无，因此当您尝试执行 np.around(None)时，脚本将不会继续执行。

调整 cv2上的 param1。HoughCircles应该可以解决问题。



HoughCircles param1 source。