在Python中使用CV2填充圆检测?
作者:互联网
我试图检测images like this.中的所有圈子
我有很多不同的图像,但是在所有圆圈中都是黑色(或几乎是黑色)并且尺寸相同(/ – 几个像素).我相信每张图片中只有2943个圆圈.这些条件永远不变.我可能无法控制图像中圆圈的大小(半径通常在15-45像素之间 – 上面提供的示例图像的半径为20-21像素).
我需要能够尽可能准确和精确地检测这些圆的中心的确切位置(如果可能的话,半径也是如此).
我尝试使用cv2.HoughCircles函数来执行此操作,但结果非常不一致且不可靠.这是我使用的代码:
from pylab import *
import sys, cv2
filename = sys.argv[1];
img = cv2.imread(filename,0);
cimg = cv2.cvtColor(img,cv2.COLOR_GRAY2BGR);
circles = cv2.HoughCircles(img,cv2.cv.CV_HOUGH_GRADIENT,2,15,param1=100,param2=30,minRadius=15,maxRadius=25);
circles = np.uint16(np.around(circles));
for i in circles[0,:]:
cv2.circle(cimg,(i[0],i[1]),i[2],(0,255,0),1);
cv2.circle(cimg,(i[0],i[1]),2,(0,0,255),3);
cv2.imwrite('1-%s'%(filename),cimg)
print "%d circles found."%(len(circles[0]));
结果是this image,输出结果为:找到了2806个圆圈.
有许多虚假圈子,许多真实的圈子被遗漏/忽略.
我开始相信如果我的所有圆圈在单个图像中是相同的,那么HoughCircle方法不是最佳的方法,并且可能有一些更好的物体检测方法可用.
如果我可以足够严格地控制圆圈的属性,你建议我用什么来精确准确地检测数千个图像中的每个圆圈?
解决方法:
我想出了这个代码,它被调整到你提供的精确图像,也找到了2943个半径估计的圆圈,假设所有的圆都有相同的半径.这是它产生的(裁剪,原始太大):
你可以看到它并不完全理想(角圆有点偏离).
它基于阈值处理,然后是轮廓运算而不是霍夫圆.
import cv2
import numpy as np
original = cv2.imread("test.jpg", cv2.CV_LOAD_IMAGE_GRAYSCALE)
retval, image = cv2.threshold(original, 50, 255, cv2.cv.CV_THRESH_BINARY)
el = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
image = cv2.dilate(image, el, iterations=6)
cv2.imwrite("dilated.png", image)
contours, hierarchy = cv2.findContours(
image,
cv2.cv.CV_RETR_LIST,
cv2.cv.CV_CHAIN_APPROX_SIMPLE
)
drawing = cv2.imread("test.jpg")
centers = []
radii = []
for contour in contours:
area = cv2.contourArea(contour)
# there is one contour that contains all others, filter it out
if area > 500:
continue
br = cv2.boundingRect(contour)
radii.append(br[2])
m = cv2.moments(contour)
center = (int(m['m10'] / m['m00']), int(m['m01'] / m['m00']))
centers.append(center)
print("There are {} circles".format(len(centers)))
radius = int(np.average(radii)) + 5
for center in centers:
cv2.circle(drawing, center, 3, (255, 0, 0), -1)
cv2.circle(drawing, center, radius, (0, 255, 0), 1)
cv2.imwrite("drawing.png", drawing)
希望能帮助到你
标签:python,opencv,computer-vision,circle,feature-detection 来源: https://codeday.me/bug/20190725/1531619.html