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