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

Improved YOLOv7-tiny algorithm for surface defect detection of bushing parts

  • 摘要: 为了能够提高轴套表面缺陷的检测精度和效率,文章提出了改进YOLOv7-tiny的轴套表面缺陷检测算法。首先在模型的特征提取上,针对处理任意维度的数据,把标准卷积替换为全维动态卷积(omni dimensional dynamic convolution,ODConv);其次在特征融合中,把上采样部分的最邻近插值替换为轻量级算子CARAFE;在拼接处引入BiFormer,增加对局部小目标的检测;最后通过把标准卷积替换为GSConv的方式,引入Slim-Neck模块。最终,在轴套数据集上做对比和消融实验,与原模型相比,改进后的算法在mAP上提高了7.7%,在局部小目标上提高了11%;在FPS上提升了40.3。用改进后的算法在公开GC10-DET数据集下做通用性实验,结果表明该算法具有通用性。

     

    Abstract: 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.

     

/

返回文章
返回