New website getting online, testing
    • 摘要: 针对复杂背景下焊接缺陷特征不明显、背景信息复杂,导致传统缺陷检测算法在实际应用中精度不佳,特征丢失等问题,本文提出一种改进自YOLOv8的焊缝表面缺陷检测算法(GD-YOLO)。模型首先引进特征提取模块与卷积模块融合,增强模型信息的提取能力;然后在颈部网络中嵌入Slim-Neck结构并在特征融合阶段引用上采样算子CAFARE,辅助增强模型性能;其次,改进注意力机制模块,使之在不显著增加计算负担的情况下,优化整体性能;最后,改用损失函数Inner-SIoU,解决边界框不匹配的问题。实验结果表明,本文模型mAP0.5检测指标比基线模型高7.8%,参数量和计算量分别减少了0.12 M、0.7 G。

       

      Abstract: In response to the problems of traditional defect detection algorithms, such as poor accuracy and feature loss in practical applications due to the inconspicuous characteristics of welding defects and complex background information, this paper proposes a welding surface defect detection algorithm based on the improved YOLOv8 (GD-YOLO). The model first introduces the fusion of feature extraction modules and convolutional modules to enhance its information extraction capabilities. Then, a slim-neck structure is embedded in the neck network, and the upsampling operator CAFARE is referenced in the feature fusion stage to assist in enhancing the model's performance. Subsequently, the attention mechanism module is improved to optimize the overall performance without significantly increasing the computational burden. Finally, the loss function is changed to Inner-SIOU to address the problem of mismatched bounding boxes. Experimental results show that the mAP0.5 detection metric of the model in this paper is 7.8% higher than that of the baseline model, and the number of parameters and the amount of computation are reduced by 0.2 M and 0.7 G, respectively.