识别附件一图片椭圆中的圆形个数;效果类似附件二;请问可以用python-opencv的 HoughCircles方法实现吗?

wangxi得 发布于 04/19 15:54
阅读 339
收藏 0

开源之夏第三届火热来袭,高校学生参与赢万元奖金!>>>

附件一:

 

附件二:

3.实现代码:

# 加载环境包
import cv2
import numpy as np

# 图片简单处理
img = cv2.imread('./file/test.jpg')  # 读取图片
GrayImage = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)  # 灰度化
GrayImage = cv2.medianBlur(GrayImage, 5)  # 中值模糊

# 阈值处理,输入图片默认为单通道灰度图片
ret, th1 = cv2.threshold(GrayImage, 127, 255, cv2.THRESH_BINARY)  # 固定阈值二值化
# threshold为固定阈值二值化
# 第二参数为阈值
# 第三参数为当像素值超过了阈值(或者小于阈值,根据type来决定),所赋予的值(一般情况下,都是256色,所以默认最大为255# thresh_binary是基于直方图的二值化操作类型,配合threshold一起使用。此外还有cv2.THRESH_BINARY cv2.THRESH_BINARY_INV cv2.THRESH_TRUNC cv2.THRESH_TOZEROcv2.THRESH_TOZERO_INV
th2 = cv2.adaptiveThreshold(GrayImage, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, 3, 5)
# adaptiveThreshold自适应阈值二值化,自适应阈值二值化函数根据图片一小块区域的值来计算对应区域的阈值,从而得到也许更为合适的图片。
# 第二参数为当像素值超过了阈值(或者小于阈值,根据type来决定),所赋予的值(一般情况下,都是256色,所以默认最大为255# 第三参数为阈值计算方法,类型有cv2.ADAPTIVE_THRESH_MEAN_Ccv2.ADAPTIVE_THRESH_GAUSSIAN_C
# 第四参数是基于直方图的二值化操作类型,配合threshold一起使用。此外还有cv2.THRESH_BINARY cv2.THRESH_BINARY_INV cv2.THRESH_TRUNC cv2.THRESH_TOZEROcv2.THRESH_TOZERO_INV
# 第五参数是图片中分块的大小
# 第六参数是阈值计算方法中的常数项
th3 = cv2.adaptiveThreshold(GrayImage, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 3, 5)
# 同上
kernel = np.ones((5, 5), np.uint8)  # 创建全一矩阵,数值类型设置为uint8
erosion = cv2.erode(th2, kernel, iterations=1)  # 腐蚀处理
dilation = cv2.dilate(erosion, kernel, iterations=1)  # 膨胀处理

imgray = cv2.Canny(erosion, 30, 100)  # Canny算子边缘检测

circles = cv2.HoughCircles(imgray, cv2.HOUGH_GRADIENT,
                           1.5,
                           100,
                           param1=90,
                           param2=20,
                           minRadius=30,
                           maxRadius=50)  # 霍夫圆变换
# 3参数默认为1
# 4参数表示圆心与圆心之间的距离(太大的话,会很多圆被认为是一个圆)
# 5参数默认为100
# 6参数根据圆大小设置(圆越小设置越小,检测的圆越多,但检测大圆会有噪点)
# 7圆最小半径
# 8圆最大半径
circles = np.uint16(np.around(circles))
# np.uint16数组转换为16位,0-65535
# np.around返回四舍五入后的值

P = circles[0]  # 去掉circles数组一层外括号
for i in P:
    # 画出外圆
    cv2.circle(img, (i[0], i[1]), i[2], (0, 255, 0), 2)  # 第二参数()内是圆心坐标,第三参数是半径,第四参数()内是颜色,第五参数是线条粗细
    # 画出圆心
    cv2.circle(img, (i[0], i[1]), 2, (0, 0, 255), 3)
print("圆的个数是:")
print(len(P))
for i in P:
    r = int(i[2])
    x = int(i[0])
    y = int(i[1])
    print("圆心坐标为:", (x, y))
    print("圆的半径是:", r)

cv2.imshow('detected circles', img)  # 第一参数为窗口名称

cv2.waitKey(0)  # 无穷大等待时间
cv2.destroyAllWindows()

运行后效果不理想。请问如何设置HoughCircles参数还是换为别的方法实现?

 

 

 

加载中
OSCHINA
登录后可查看更多优质内容
返回顶部
顶部