结合注意力机制与自适应记忆性融合网络的纺织品瑕疵检测

摘要：

为了解决纺织生产工艺中瑕疵检测成本较高、精度较低、速度较慢等问题,文中提出结合注意力机制与自适应记忆性融合网络的纺织品瑕疵检测模型.首先,在YOLOv5骨干网络中引入改进的注意力模块,构建特征提取网络,增强模型对纺织品瑕疵特征的提取能力.然后,为了增强浅层定位信息的传递效应和有效缓解特征融合时产生的混叠效应,提出自适应记忆性融合网络,在提高特征尺度不变性的同时,将骨干网络中的特征信息融入特征融合层.最后,引入CDIoU(Control Distance Intersection over Union)损失函数,提高检测精度.在ZJU-Leaper纺织品瑕疵数据集和天池纺织品瑕疵数据集上的实验表明,文中模型具有较高的检测精度和较快的检测速度.

关键词：纺织品瑕疵检测、注意力机制、YOLOv5、自适应记忆性融合网络

所属期刊栏目：35

分类号：TP391(计算技术、计算机技术)

在线出版日期：2022-07-14（万方平台首次上网日期，不代表论文的发表时间）

页数：共12页

页码：536-547