用python实现开放式cv中的圆检测

Question

我试着用红色的圆形物体来探测黑色背景中的圆形。

import cv2
import cv2.cv as cv
import numpy as np

img = cv2.imread('extracted.jpg',0)
img = cv2.medianBlur(img,5)
cimg = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

circles = cv2.HoughCircles(img,cv.CV_HOUGH_GRADIENT,1,20,
                        param1=50,param2=30,minRadius=0,maxRadius=0)

circles = np.uint8(np.around(circles))
for i in circles[0,:]:
   # 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.imshow('detected circles',cimg)
cv2.waitKey(0)
cv2.destroyAllWindows()

我已经在灰度模式下加载了图像，但它仍然会给我一个错误

"circles = np.uint8(np.around(circles))
File "/usr/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 2277, in around
  return _wrapit(a, 'round', decimals, out)
File "/usr/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 37, in _wrapit
  result = getattr(asarray(obj),method)(*args, **kwds)
AttributeError: rint"

因为我现在的名声，我不能张贴这张照片。

Answer 1

在你的代码中有一个小的修正。

您正在以灰度加载图像，然后再次使用无效操作cv2.cvtColor将其转换为灰度。

或者，OpenCV提供一个sample for circle detection using Hough Circles method。你可以试试。

如果您使用的是OpenCV 2.x版本，只需将cv2.LINE_AA更改为cv2.CV_AA或您喜欢的任何其他线型。