基于改进YOLOv5s的粉末冶金工件表面缺陷检测

作     者：万霞 赵为鑫 张文超 王晨曦 王明伟 姜瀚 

作者机构：大连工业大学机械工程与自动化学院辽宁大连116034 科斯特数字化智能科技(深圳)有限公司广东深圳518101 

基　　金：辽宁“百千万人才工程”培养经费资助(无编号) 

出 版 物：《科技创新与应用》 (Technology Innovation and Application)

年 卷 期：2023年第13卷第24期

页      码：72-77页

摘      要：针对粉末冶金工件表面发生开裂、孔洞等缺陷时,采用机器视觉检测易出现漏检和误检等问题,提出一种改进的YOLOv5s目标检测算法。综合考虑工件表面缺陷的特点,采用线性卷积替换普通卷积降低网络模型参数,在特征提取网络中引入坐标注意力机制提升目标通道和位置通道特征能力,重新设计先验框尺寸提高锚框尺寸匹配度,并对损失函数进行改进,进而提升检测精度。实验表明,改进后的网络平均检测精度(mAP)由原来的58.5%提升至63.6%,检测速度达到65 ms。改进后的算法在YOLOv5s轻量高效的前提下,更能实现精准的零件缺陷检测。

主 题 词：YOLOv5算法 注意力机制 GhostNet 表面缺陷检测 锚框 

学科分类：08[工学] 080502[080502] 0805[工学-能源动力学] 

D　O　I：10.19981/j.CN23-1581/G3.2023.24.015

馆 藏 号：203123106...