Abstract:
Castings\' surface-defect detection is a crucial machine vision-based automation technology. This paper proposes a fusion-enhanced attention mechanism and efficient self-architecture lightweight YOLO (SLGA-YOLO) to overcome the existing target detection algorithms\' poor computational efficiency and low defect-detection accuracy. We used the SlimNeck module to improve the neck module and reduce redundant information interference. The integration of simplified attention module (SimAM) and Large Separable Kernel Attention (LSKA) fusion strengthens the attention mechanism, improving the detection performance, while significantly reducing computational complexity and memory usage. To enhance the generalization ability of the model\'s feature extraction, we replaced part of the basic convolutional blocks with the self-designed GhostConvML (GCML) module, based on the addition of p2 detection. We also constructed the Alpha-EIoU loss function to accelerate model convergence. The experimental results demonstrate that the enhanced algorithm increases the average detection accuracy (mAP@0.5) by 3% and the average detection accuracy (mAP@0.5:0.95) by 1.6% in the castings\' surface defects dataset.
摘要:
铸件表面缺陷检测是基于机器视觉的重要自动化技术。本文提出了一种融合增强的注意力机制和高效的自架构轻量级YOLO(SLGA-YOLO),以克服现有目标检测算法的计算效率低下和缺陷检测精度低下的问题。我们使用SlimNeck模块来改进颈部模块并减少冗余信息干扰。简化注意力模块(SimAM)和大型可分离内核注意力(LSKA)融合的集成加强了注意力机制,提高检测性能,同时显著降低计算复杂度和内存使用。为了增强模型特征提取的泛化能力,我们用自行设计的GhostConvML(GCML)模块替换了部分基本卷积块,基于添加p2检测。我们还构造了Alpha-EIoU损失函数来加速模型的收敛。实验结果表明,在铸件表面缺陷数据集中,增强算法的平均检测精度(mAP@0.5)提高了3%,平均检测精度(mAP@0.5:0.95)提高了1.6%。