基于改进YOLOv11的活塞角部缺陷检测模型

An Improved YOLOv11-Based Model for Piston Corner Defect Detection

  • 摘要: 在工业制造场景下,活塞角部的微小缺陷若无法被及时、准确地识别,往往会对设备运行性能乃至整体安全性造成潜在威胁。针对目前缺陷目标尺度极小、成像环境复杂多变且金属表面易产生强反光的问题,本文提出了一种轻量化模型MDE-YOLO。通过引入非扩张多尺度卷积(PKI)以强化局部细节与多尺度特征表征,并利用动态特征融合模块(MFM)自适应调节跨尺度特征的权重分配;同时,在检测头部分融合MBConv结构与SE通道注意力机制,在保证特征提取能力的前提下有效抑制计算开销。在自构建的活塞缺陷数据集上,MDE-YOLO实现了96.6%的mAP@0.5,计算复杂度约为5.2 GFLOPs、推理速度达到182.7 FPS,在仅有2.3 M参数量的情况下其效果仍优于多种主流检测模型;进一步在MS COCO数据集上的评估也显示,该模型在复杂场景下保持了良好的泛化性能与应用价值,适用于对实时性与检测精度均具有较高要求的工业缺陷检测任务。

     

    Abstract: In industrial manufacturing, tiny defects at piston corners can cause serious problems if they are not detected accurately and in time. Such defects may affect equipment performance and even threaten operational safety. However, defect detection remains challenging due to very small target sizes, complex imaging conditions, and strong reflections on metal surfaces. To address these issues, this paper proposes a lightweight model named MDE-YOLO. First, a non-dilated multi-scale convolution module (PKI) is introduced to enhance local details and multi-scale feature representation. Second, a dynamic feature fusion module (MFM) is used to adaptively adjust the weights of features from different scales. In addition, the detection head integrates MBConv structures and SE channel attention, which improves feature extraction while keeping computational cost low. Experiments on a self-built piston defect dataset show that MDE-YOLO achieves 96.6% mAP@0.5, with a computational cost of about 5.2 GFLOPs and an inference speed of 182.7 FPS. The model has only 2.3M parameters, yet it outperforms many mainstream detection models. Further evaluation on the MS COCO dataset demonstrates that MDE-YOLO maintains good generalization ability in complex scenes. These results indicate that the proposed method is well suited for industrial defect detection tasks that require both high accuracy and real-time performance.

     

/

返回文章
返回