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

摘要

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