当前位置:主页 > 软件编程 > Python代码 >

如何基于OpenCV&Python实现霍夫变换圆形检测

时间:2022-10-20 10:15:16 | 栏目:Python代码 | 点击:

简述

基于python使用opencv实现在一张图片中检测出圆形,并且根据坐标和半径标记出圆。不涉及理论,只讲应用。

霍夫变换检测圆形的原理

其实检测圆形和检测直线的原理差别不大,只不过直线是在二维空间,因为y=kx+b,只有k和b两个自由度。而圆形的一般性方程表示为(x-a)²+(y-b)²=r²。那么就有三个自由度圆心坐标a,b,和半径r。这就意味着需要更多的计算量,而OpenCV中提供的cvHoughCircle()函数里面可以设定半径r的取值范围,相当于有一个先验设定,在每一个r来说,在二维空间内寻找a和b就可以了,能够减少计算量。

相关函数

函数说明:

Python: cv2.HoughCircles(image, method, dp, minDist[, circles[, param1[, param2[, minRadius[, maxRadius]]]]]) → circles

参数说明:

这是根据opencv官方文档谷歌翻译过来的,参数比较多,但用的时候只修改一些主要的,传入的图像和最大最小圆半径,以达到检测出想要的圆的效果。

还有要注意函数的返回值

找到的圆的输出向量。每个向量被编码为3元素的浮点向量 (x,y,半径)。

这句话不是很好理解,我们直接输出返回值就会发现是个 三层嵌套list,最内层list有三个元素,分别是圆心的行,列,半径,这表示一个圆的基本信息。多个圆基本信息组成了一个新的list,这个list包含了检测到的所有圆,即长度就是检测到圆的个数。最外层再加了个list,至于干啥的不清楚了,不影响使用就好。

下图是输出函数返回值的一个实例图

实例演示

完整代码

import  cv2

#载入并显示图片
img=cv2.imread('circle.png')
cv2.imshow('img',img)
#灰度化
gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
#输出图像大小,方便根据图像大小调节minRadius和maxRadius
print(img.shape)
#霍夫变换圆检测
circles= cv2.HoughCircles(gray,cv2.HOUGH_GRADIENT,1,100,param1=100,param2=30,minRadius=5,maxRadius=300)
#输出返回值,方便查看类型
print(circles)
#输出检测到圆的个数
print(len(circles[0]))

print('-------------我是条分割线-----------------')
#根据检测到圆的信息,画出每一个圆
for circle in circles[0]:
    #圆的基本信息
    print(circle[2])
    #坐标行列
    x=int(circle[0])
    y=int(circle[1])
    #半径
    r=int(circle[2])
    #在原图用指定颜色标记出圆的位置
    img=cv2.circle(img,(x,y),r,(0,0,255),-1)
#显示新图像
cv2.imshow('res',img)

#按任意键退出
cv2.waitKey(0)
cv2.destroyAllWindows()

运行结果

shell输出截图

总结

您可能感兴趣的文章:

相关文章