之前在准备中国机器人大赛的微型无人机赛道时,要求识别以下标靶中正中央的内容。
由于带着外圆可能会产生误识别导致识别错误,于是想要使用霍夫圆检测的方法定位图像中间位置,并对图像进行裁剪输出合适的ROI区域。
理论上来说霍夫圆检测算法能识别出图中的两个圆,只需要使用其中一个圆的半径大小进行图像的裁剪即可。但是在实际操作过程中可能有时只会识别出其中一个圆,这样就会导致实际切割出的ROI区域不符合所需要求。
针对以上问题,解决方案是先将图片进行二值化,去除其中的一个圆,接下来再对二值化后的图像进行霍夫圆检测,这样就不会因为检测出的圆的半径不时跳动而造成识别上的困难。
以下分别是原始输入图像、二值化后图像、Canny算子图像以及最终裁剪图像。
可以看出,通过这种二值化后进行检测方式裁剪出来的图像准确性比直接使用霍夫圆检测算法具有更高的稳定性以及准确性。
以下是二值化之后再进行霍夫圆检测的代码。代码先将原始图像转化为灰度图,再进行二值化处理,经过一次高斯滤波后使用Canny算子寻找圆轮廓,接着再进行一次高斯滤波,接下来进行霍夫圆检测即可。
R_Parameter = 0.45
def circlecut(frame):
if(frame is not None):
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) # 转换为灰色通道
r = 0
# 消除噪声
thresh, binary = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY) #全局二值化
Gauss1 = cv2.GaussianBlur(binary, (5, 5), 2, 1)
edges = cv2.Canny(Gauss1, 50, 100) # 边缘识别
Gauss2 = cv2.GaussianBlur(edges, (5, 5), 2, 1)
cv2.imshow('Gauss2', Gauss2)
# 识别圆形
circles = cv2.HoughCircles(Gauss2, cv2.HOUGH_GRADIENT, 1, 10, param1=1, param2=80, minRadius=20, maxRadius=200)
if circles is not None: # 如果识别出圆
for circle in circles[0]:
# 获取圆的坐标与半径
if(int(circle[2]) > r):
x = int(circle[0])
y = int(circle[1])
r = int(circle[2])
if((y-r*R_Parameter) > 0 and (y+r*R_Parameter) > 0 and (x-r*R_Parameter) < 480 and (x+r*R_Parameter) < 640): #防止ROI区域大于屏幕区域
frame = frame[int(y-r*R_Parameter):int(y+r*R_Parameter), int(x-r*R_Parameter):int(x+r*R_Parameter)]
frame = cv2.resize(frame, (128, 128), interpolation = cv2.INTER_AREA) #改变图片长宽
return frame
else:
return None
其中需要注意的点是,不知为何,使用其它的滤波算法或者不进行滤波会导致霍夫圆检测失败,测试时只有高斯滤波之后才能使用。