首页|基于YOLOv5算法的圆心定位方法

基于YOLOv5算法的圆心定位方法

扫码查看
准确地获取图像中圆形目标的圆心是目标识别和定位中的关键问题.目前圆心定位主要采用最小二乘拟合圆以及HOUGH变换方法,但这些方法在不同程度上存在着鲁棒性不强、对环境光线要求高、需提前调试参数确定阈值、复杂背景下效果急剧变差等局限性.针对该问题,提出一种综合运用YOLOv5算法、Grabcut算法和灰度质心法进行图像分类、图像分割和灰度重心算法求取圆心坐标.该方法使用YOLOv5对图像中的圆形目标进行粗定位,再通过图像分割方法分割出圆形目标,最后使用加权型的灰度重心算法准确定位出圆心坐标,实现对圆形标志的可靠定位.用多个实验比较了该算法与现有算法的精度和稳定性,实验表明,该算法与现有算法相比,在干扰及形变的情况下圆心定位平均相对误差保持在0.5 pixel以内,而在干扰、强形变及目标残缺的情况下仍能保持在7 pixel以内,该算法不仅提高了圆形目标圆心检测的鲁棒性和准确性,而且具有良好的抗干扰性.
A Circle Center Location Method Based on YOLOv5 Algorithm
Accurately obtaining the center of a circular object in an image is a key issue in object recognition and local-ization.At present,the least square fitting circle and HOUGH transform methods are mainly used to locate the center of a circle.However,these methods have limitations in varying degrees,such as weak robustness,high requirements for ambient light,the need to debug parameters in advance to determine the threshold value,and sharp deterioration of the effect un-der complex background.In response to this issue,this paper proposes a comprehensive application of YOLOv5 algorithm,Grabcut algorithm,and grayscale centroid method for image classification,image segmentation,and grayscale centroid al-gorithm to obtain the center of circle coordinates.This method uses YOLOv5 to coarse locate circular targets in the image,then segments circular targets through image segmentation methods.Finally,a weighted grayscale center of gravity algo-rithm is used to accurately locate the center coordinates of the circle,achieving reliable positioning of circular markers.This paper compares the accuracy and stability of our algorithm with existing algorithms through multiple experiments.

center positioningYOLOv5 algorithmimage segmentationgrayscale centroid method

肖钦峰、魏东、刘波、莫永迪、王炳智

展开 >

北华航天工业学院电子与控制工程学院,河北廊坊 065000

圆心定位 YOLOv5算法 图像分割 灰度质心法

2024

工业控制计算机
中国计算机学会工业控制计算机专业委员会 江苏省计算技术研究所有限责任公司

工业控制计算机

影响因子:0.258
ISSN:1001-182X
年,卷(期):2024.37(1)
  • 10