基于YOLOv8的微小缺陷检测算法研究
Research on Micro Defect Detection Algorithm Based on YOLOv8
摘要: 工业产品质量是智能制造的核心竞争力,表面缺陷检测作为质检关键环节,直接决定生产效率与产品良率。针对传统检测方法在复杂工业场景中存在的精度不足、泛化能力弱及模型冗余等痛点,本研究构建多策略协同优化的YOLOv8模型,通过三重特征优化、创新上采样方法及系统级开发,形成高精度、轻量化检测解决方案,为智能制造提供技术支撑。具体研究内容如下:一,构建多维度协同优化YOLOv8模型,采用卷积核感受野调控、可变形注意力机制引入及复合损失函数设计的三重策略提升特征提取效能,在MVTec AD Dataset上平均精度较原始模型提升5.2%,实现计算量与精度的平衡及梯度更新的精准性;二,提出通道独立上采样模块,通过单通道独立重建、渐进式上采样及轻量化全局感知模块,解决语义衰减难题,在NEU-DET、铝型材表面检测数据集(APDDD)等工业标准数据集上稳定提升检测精度并控制成本,具备多工具适配及结果可视化分析能力,可降低企业二次开发成本,提供工程化实时检测方案。
Abstract: Industrial product quality is the core competitiveness of intelligent manufacturing, and surface defect detection, as a key link in quality inspection, directly affects production efficiency and product yield. In view of the accuracy deficiency, weak generalization ability and model redundancy of traditional detection methods in complex industrial scenarios, this study constructs a YOLOv8 model with multi-strategy collaborative optimization, and through triple feature optimization, an innovative up-sampling method and system-level development, it forms a high-precision and lightweight detection solution providing technical support for intelligent manufacturing. The specific research content is as follows: First, a multi-dimensional collaborative optimization YOLOv8 model is constructed, and a triple strategy convolutional kernel receptive field regulation, deformable attention mechanism introduction and composite loss function design is adopted to improve the efficiency of feature extraction. The average accuracy on MVTec AD Dataset is 5.2% higher than that of the original model, achieving a balance between computational amount and accuracy, as well as the precision of gradient update; Second, a channel-independent-sampling module is proposed. Through single-channel independent reconstruction, progressive up-sampling and a lightweight global perception module, the problem of semantic attenuation is solved. The detection accuracy is improved on industrial standard datasets such as NEU-DET and the Aluminum Profile Surface Inspection Dataset (APDDD), and the cost is controlled. It has the features of multi-tool adaptation and result visualization analysis, which can reduce the cost of secondary development of enterprises and provide engineering real-time detection solutions.
参考文献
|
[1]
|
Szeliski, R. (2022) Computer Vision: Algorithms and Applications. 3rd Edition, Springer. [Google Scholar] [CrossRef]
|
|
[2]
|
Liu, Y., Xu, K. and Wang, D.D. (2018) Online Surface Defect Identification of Cold Rolled Strips Based on Local Binary Pattern and Extreme Learning Machine. Metals, 8, Article 197. [Google Scholar] [CrossRef]
|
|
[3]
|
Breiman, L. (2001) Random Forests. Machine Learning, 45, 5-32. [Google Scholar] [CrossRef]
|
|
[4]
|
叶志晖, 武健, 赵晓忠, 等. 基于特征交互与自适应分组融合的多模态目标检测[J]. 红外技术, 2025, 47(4): 468-474.
|
|
[5]
|
陈嘉欣, 孙传猛, 葛耀栋. 基于CycleGAN和Pix2Pix的轨面缺陷图像智能生成技术[J]. 铁道学报, 2024, 46(2): 122-130.
|
|
[6]
|
Chen, Z., Qin, W., He, G., Li, J., Huang, R., Jin, G., et al. (2023) Explainable Deep Ensemble Model for Bearing Fault Diagnosis under Variable Conditions. IEEE Sensors Journal, 23, 17737-17750. [Google Scholar] [CrossRef]
|
|
[7]
|
曹家乐, 李亚利, 孙汉卿, 等. 基于深度学习的视觉目标检测技术综述[J]. 中国图象图形学报, 2022, 27(6): 1697-1722.
|
|
[8]
|
孙钰, 严宇, 崔剑, 等. 联邦学习深度梯度反演攻防研究进展[J]. 电子与信息学报, 2024, 46(2): 428-442.
|
|
[9]
|
Li, G., Shao, R., Wan, H., Zhou, M. and Li, M. (2022) A Model for Surface Defect Detection of Industrial Products Based on Attention Augmentation. Computational Intelligence and Neuroscience, 2022, 1-12. [Google Scholar] [CrossRef] [PubMed]
|
|
[10]
|
Zhao, W., Chen, F., Huang, H., et al. (2021) A New Steel Defect Detection Algorithm Based on Deep Learning. Computational Intelligence and Neuroscience, 2021, Article 5592878.
|