基于改进YOLOv11n的螺栓缺陷检测方法

Bolt defect detection method based on improved YOLOv11n

  • 摘要: 螺栓作为常见紧固件广泛应用于多种场合,针对螺栓目标小、复杂背景干扰引起的特征提取不充分和检测精度下降问题,提出一种改进的YOLOv11n螺栓缺陷检测方法。首先,为强化模型对关键特征的捕捉能力,采用卷积注意力机制模块(convolutional block attention module,CBAM)对YOLOv11n进行改进;其次,为提升特征提取环节计算效率与精度的平衡,将 MobileNetV3 网络嵌入骨干网络结构中;此外,将CIoU替换为Alpha-IoU以提升边界框回归精度;最后,针对原检测头中分类与回归任务共享特征导致的任务干扰问题,将YOLO检测头解耦。在自制数据集上进行训练和测试,实验结果表明,该方法检测精度和均值平均精度分别为88.7%和82.5%,较YOLOv11n分别提升4.3%和3.4%,提升了检测精度与复杂场景下的鲁棒性。

     

    Abstract: Bolts, as common fasteners, are widely used in various applications. To address the issues of insufficient feature extraction and decreased detection accuracy caused by small bolt targets and complex background interference, an improved YOLOv11n bolt defect detection method is proposed. Firstly, to enhance the model's feature extraction capability, the CBAM attention mechanism is introduced. Secondly, in order to improve the balance between computational efficiency and accuracy in the feature extraction phase, the MobileNetV3 network is embedded into the backbone network structure. Additionally, CIoU is replaced with Alpha-IoU to improve bounding box regression accuracy. Finally, to address task interference caused by shared features in the original detection head for classification and regression tasks, the YOLO detection head is decoupled. Training and testing are conducted on a self-made dataset. Experimental results show that this method achieves a detection accuracy and mean average precision of 88.7% and 82.5%. Respectively, representing improvements of 4.3% and 3.4% compared to YOLOv11n, enhancing detection accuracy and robustness in complex scenarios.

     

/

返回文章
返回