我们的服务符合学术规范和道德
专业 高端让您使用时没有后顾之忧

机械传动2024年第05期:基于BN-YOLOv5的轻量级齿轮表面缺陷检测方法

来源:期刊VIP网 时间:


作者:赵小惠;张智杰;胡胜;郇凯旋;刘磊;蒲军平;

单位:西安工程大学机电工程学院;

摘要:齿轮表面的缺陷检测是齿轮生产制造过程中相当重要的工序。为了提高齿轮表面缺陷检测的精度,提出了一种基于改进YOLOv5的算法检测模型BN-YOLOv5。首先,将加权双向特征金字塔网络结构嵌入到颈部网络结构中,强化了网络对不同特征的提取能力;其次,引入轻量级的基于标准化的注意力模块(Normalization-based Attention Module,NAM),将其与加权双向特征金字塔网络结构相结合,以更加有效地融合高层与低层的特征信息;最后,采用深度可分离卷积模块替换网络结构中所有的卷积层,使网络模型更加轻量化。实验结果显示,改进后的算法模型平均精度均值可达到98.5%,检测速度达到66 FPS/s,模型大小为9.69 MB,有效降低了模型的占用内存,可满足在小型移动设备上实时检测齿轮表面缺陷的任务要求。

关键词:齿轮表面;;缺陷检测;;YOLOv5;;轻量级;;NAM注意力机制

基金资助:国家自然科学基金项目(72001166);; 陕西省科技计划项目(2022JQ-721);; 陕西省社会科学界联合会项目(20ZD195-59)