改进HED网络在轴承自动测量中的应用研究

来源：电子测量技术杂志2024年第2期北京时间：

作者：陈世奇，侯明，李鹏程

单位：北京信息科技大学自动化学院 北京 100192

摘要：针对在工业环境中传统接触式测量轴承尺寸操作复杂且检测成本高的问题，提出一种新的加入注意力机制和结合Canny算法的改进型HED网络边缘检测算法。该方法基于HED网络，将主干网络第4、5个阶段的卷积层替换成连续空洞卷积，并设置网络的第3、4层池化步长为1，增大模型的感受野，提高输出的边缘图像精度；加入高效通道注意力机制ECA模块，有效抑制无关纹理特征和非边缘像素等影响；使用Canny算法中的非极大值抑制和双阈值处理算法，对检测的粗边缘进行细化，得到更加精确的轴承边缘；使用最小二乘圆拟合算法，获取轴承的内外环尺寸参数。实验结果表明，改进后的HED网络在ODS和OIS指标上分别达到了0.811和0.835，该方法可以有效实现轴承边缘检测并保证轴承尺寸测量精度。

关键词：HED网络;轴承边缘检测;尺寸测量;空洞卷积;注意力机制

基金资助：国家自然科学基金(61971048)项目资助