基于RSME-YOLO的轻量化LED灯泡外观缺陷检测方法

Lightweight LED bulb appearance defect detection method based on RSME-YOLO

  • 摘要: 为解决现有LED灯泡外观缺陷检测算法计算资源需求大和精度低的问题,设计一种RSME-YOLO算法。首先,提出一种重构的轻量化RD-HGNet主干,降低计算冗余的同时增强梯度稳定。其次,采用Slim-Neck模块,减少冗余计算,同时保留跨通道交互,高效维持多尺度特征融合。然后,设计一种融合多头自注意力机制的MSA-Detect检测头,增强特征交互能力和模型表达能力。最后,通过更换长宽分别解耦的EIoU损失函数,解决不平衡优化对模型带来的负面影响。与YOLOv8n基础算法相比,RSME-YOLO的参数量下降了33.2%,GFLOPs降低40.7%,权重文件大小仅为4.2 M,在验证集和测试集上mAP50分别达到87.1%和86.8%。该算法检测精度更高,同时实现了轻量化,更加适用于中小型企业对LED缺陷检测的智能升级。

     

    Abstract: To address the issues of high computational resource demands and low accuracy in existing defect detection algorithms for LED bulb appearance defect detection, an RSME-YOLO algorithm wasd designed. First, a reconstructed lightweight RD-HGNet backbone was proposed to reduce computational redundancy while enhancing gradient stability. Second, a Slim-Neck module was adopted to minimize redundant computations while preserving cross-channel interactions, efficiently maintaining multi-scale feature fusion. Then, a MSA-Detect detection head incorporating multi-head self-attention mechanism was designed to enhance feature interaction capability and model expressiveness. Finally, by replacing the loss function with the decoupled EIoU loss that separately handles length and width, the negative impact of unbalanced optimization on the model was resolved. Compared to the baseline YOLOv8n algorithm, RSME-YOLO reduced the parameter count by 33.2%, lowered GFLOPs by 40.7%, and had a model size of only 4.2 M. It achieved mAP50 scores of 87.1% and 86.8% on the validation and test sets, respectively. With higher detection accuracy and lightweight design, it is more suitable for intelligent upgrades in LED defect detection for small and medium-sized enterprises.

     

/

返回文章
返回