首页|改进YOLOv7-tiny的轴套零件表面缺陷检测算法

改进YOLOv7-tiny的轴套零件表面缺陷检测算法

扫码查看
为了能够提高轴套表面缺陷的检测精度和效率,文章提出了改进YOLOv7-tiny的轴套表面缺陷检测算法.首先在模型的特征提取上,针对处理任意维度的数据,把标准卷积替换为全维动态卷积(omni dimensional dynamic convolution,ODConv);其次在特征融合中,把上采样部分的最邻近插值替换为轻量级算子CARAFE;在拼接处引入BiFormer,增加对局部小目标的检测;最后通过把标准卷积替换为GSConv的方式,引入Slim-Neck模块.最终,在轴套数据集上做对比和消融实验,与原模型相比,改进后的算法在mAP上提高了 7.7%,在局部小目标上提高了 11%;在FPS上提升了 40.3.用改进后的算法在公开GC10-DET数据集下做通用性实验,结果表明该算法具有通用性.
Improved YOLOv7-tiny algorithm for surface defect detection of bushing parts
For improving the precision and efficiency of sleeve surfacedefect detection,an improved YOLOv7-tiny algorithm is proposed in this paper.First,in the feature extraction of the model,the standard convolution is replaced by omni dimensional dynamic convolution for processing data of any dimension,and second,in the feature fusion,the standard convolution is replaced by omni dimensional dynamic convolution,the nearest neighbor interpolation of the upper sampling part is replaced by the light-weight operator CARAFE,the BiFormer is introduced into the splice to increase the detection of local small targets,and the Slim-Neck module is introduced by replacing the standard convolution with GSConv.Finally,compared with the original model,the modified algorithm improved 7.7%on mAP,11%on local small targets and 40.3 on FPS.Experiments on GC10-DET data set show that the improved algorithm is universal.

defect detectionomni dimensional dynamic convolutionCARAFEBiFormerSlim-Neck

李大伟、孙一兰、王品、叶明亮

展开 >

沈阳理工大学机械工程学院,辽宁 沈阳 110159

沈阳中科数控技术股份有限公司,辽宁 沈阳 110168

缺陷检测 全维动态卷积 CARAFE BiFormer Slim-Neck

国家科技重大专项

2018ZX04035001

2024

制造技术与机床
中国机械工程学会 北京机床研究所

制造技术与机床

CSTPCD北大核心
影响因子:0.264
ISSN:1005-2402
年,卷(期):2024.(6)