LLM2D
SPFFNet:条纹感知和特征融合的空间金字塔池化在织物缺陷检测中的应用
SPFFNet: Strip Perception and Feature Fusion Spatial Pyramid Pooling for Fabric Defect Detection
作者: Peizhe Zhao
发布日期: 2/4/2025
arXiv ID: oai:arXiv.org:2502.01445v1

摘要

arXiv:2502.01445v1 交叉类型: cross 摘要:织物缺陷检测对于质量控制至关重要,但现有的方法往往难以应对复杂的背景和形状特定的缺陷。本文提出一种基于YOLOv4的改进型织物缺陷检测模型。为增强对于条状缺陷的检测,我们引入了一种条状感知模块(Strip Perception Module, SPM),通过多尺度卷积提高特征捕捉能力。为进一步增强空间金字塔池化快速版(SPPF),我们整合了挤压与激励机制,形成了SE-SPPF模块,更好地结合了空间和通道信息,以实现更有效的缺陷特征提取。此外,我们提出了一种新颖的自适应权重强化完整交并比(Focal Enhanced Complete Intersection over Union, FECIoU)度量,通过调整焦损失的权重来解决比例差异和类别不平衡的问题。实验结果表明,我们的模型在天池数据集上的均值平均精度(mAP)提高了0.8-8.1%,在我们的自定义数据集上的改进幅度为1.6-13.2%,超过了现有的其他先进方法。