分享自:

基于迁移学习和改进Faster R-CNN的织物疵点检测

期刊:Journal of Engineered Fibers and FabricsDOI:10.1177/15589250221086647

文档类型判断:类型a 本研究报告的是一项单一的原创性研究,因此以下报告将按照类型a的要求进行组织。


基于迁移学习与改进Faster R-CNN的织物疵点检测研究报告

一、研究概述与发表信息

本研究由来自武汉纺织大学(Wuhan Textile University)机械工程与自动化学院、湖北省数字化纺织装备重点实验室的赵佳(Zhao Jia)、周石(Zhou Shi)、梅顺齐(Mei Shunqi)以及俄罗斯阿尔泰国立技术大学(Altai State Technical University)的郑权(Zheng Quan)等学者共同完成。通讯作者为梅顺齐。该研究成果于2022年发表在期刊《Journal of Engineered Fibers and Fabrics》上。

二、研究背景与目的

在纺织工业中,织物疵点(fabric defects)的自动检测是质量控制的关键环节,但传统的人工检测成本高、效率低,且长时间工作会损害视力。现有的自动化检测设备因知识产权限制,其算法细节不公开,且对不同类型织物的适应能力(adaptive abilities)较弱。已有的基于深度学习的检测方法虽取得了一定进展,但仍面临三大问题:其一,网络结构复杂导致训练时间长;其二,对于织物中大量存在的小目标疵点(small target defects),检测效果和精度不佳;其三,织物疵点样本数据集稀缺且多样性不足。

为解决上述问题,本研究旨在提出一种基于迁移学习(Transfer Learning)和改进的快速区域卷积神经网络(Faster R-CNN)的织物疵点检测系统,以期实现高精度、快速收敛的检测,尤其是提升对小目标疵点的识别能力。

三、研究方法与详细工作流程

本研究的方法流程主要包含以下几个步骤:数据集预处理、算法模型改进、模型训练与验证。

1. 数据集构建与预处理 研究团队从浙江省的一家纺织厂实际生产场景中采集了总计2119张织物图像,涵盖了无纹理坯布和格子坯布,颜色包括黄色、黑色等多种。图像中包含六种常见的疵点类型:带状纱(ribbon yarn)、断纱(broken yarn)、棉球(cotton ball)、破洞(hole)、脱纱(yarn shedding)和污渍(stain),每种疵点约有300张样本。所有图像分辨率统一为512×512像素,并由经验丰富的验布工进行手工分类,随后使用LabelImg工具对疵点进行标注,生成包含类别和位置信息的XML文件。为解决样本不足和多样性有限的问题,研究者对数据集进行了数据增强(data augmentation),通过镜像、旋转、曝光等操作将图像样本数扩展至6357张。最终,这个增强后的数据集被按照1:1的比例划分为训练集和验证集,用于导入改进模型进行训练。

2. 算法改进与模型设计 本研究以Faster R-CNN为基本框架,针对原始算法的不足和织物疵点检测的特性,进行了四项关键改进:

  • 特征提取网络的替换: 原始网络使用结构较深的VGG16作为特征提取网络,这会导致训练时间长、参数调整困难,并可能引发梯度消失(gradient vanishing)问题。本研究将其替换为ResNet50网络。ResNet50通过其核心的残差学习结构(residual learning structure),将堆叠的网络层视为恒等映射(identity map),有效地解决了深度网络中的退化现象(degradation phenomenon),在保证检测精度的同时,减少了网络参数和训练复杂度。
  • 引入特征金字塔网络(FPN)并与区域建议网络(RPN)融合: 原始Faster R-CNN的RPN仅作用于深层特征图上,虽然语义信息丰富,但分辨率低,不利于小目标检测。本研究引入了特征金字塔网络(FPN),将包含丰富细节信息的浅层特征与高层语义特征相连接,形成了一个多尺度的特征结构。同时,为了适应数据集中不同尺寸的疵点,RPN中的锚框(anchor box)被扩展为32、64、128、256和512五种尺寸。将FPN与扩展后的RPN融合后,网络能够在提取丰富语义信息的同时保留更多细节,从而显著提升了对小目标疵点的定位和检测精度。
  • 兴趣区域池化方式的替换: 原始Faster R-CNN中的兴趣区域池化(ROI Pooling)层在操作过程中会进行两次量化(quantization),这导致候选框(candidate box)相对于初始位置产生偏差,影响检测的精确度。为解决此问题,本研究改用兴趣区域对齐(ROI Align)方法。ROI Align取消了量化操作,采用双线性插值(bilinear interpolation)来计算浮点坐标像素的图像值,从而使整个特征聚合过程变得连续,避免了细节信息的丢失,实现了对候选框的精准定位。
  • 集成级联模块(Cascade Module): 为更精确地区分前景和背景,避免在IoU阈值(Intersection over Union)设置不当时出现冗余干扰框或模型过拟合,本研究在Faster R-CNN中引入了多层级联模块。该模块各级设定逐层递增的IoU阈值(例如,第一层为0.5,第二层为0.6,第三层为0.7),并将前一级的输出作为后一级的输入。通过三个阶段的级联、分类与回归的迭代优化,使得每个阶段专注于特定IoU阈值下的检测器训练,特征提取能力逐步增强,最终能更准确地定位不同尺寸的目标。最终的三层级联模块在保证检测速度几乎不变的情况下,使平均精度均值(mAP)提升了2.4%。
  • 应用迁移学习: 为了在数据集相对较小的情况下快速训练出理想的模型,本研究采用了迁移学习策略。具体来说,是加载了ResNet50模型在大规模数据集ImageNet上的预训练权重(pre-trained weights)来初始化改进后的网络参数,然后仅对顶层的全连接层等参数利用织物疵点数据集进行微调(fine-tuning)。这个方法极大地缩短了训练时间,并提高了模型的泛化能力。

3. 实验设置与评价指标 实验在Windows 10操作系统下,基于PyTorch和TensorFlow深度学习框架进行。硬件环境为Intel® Core™ i7-10750H CPU。研究选用了平均精度均值(mean average precision, mAP)、验证准确率、F1分数和损失值(loss)作为主要评价指标,并将模型训练速度(秒/张)作为重要参考。

四、主要研究结果

实验通过与不同配置的Faster R-CNN算法以及其他主流算法进行比较,验证了所提模型的有效性。

  • 与不同配置Faster R-CNN的性能对比: 实验中构建了多个对比模型,包括基于VGG16且引入ROI Align的Faster R-CNN (1)、基于MobileNet的Faster R-CNN (2)、基于ResNet101的Faster R-CNN (3)、基于ResNet50的Faster R-CNN (4),以及基于ResNet50并引入FPN和ROI Align的Faster R-CNN (5)。

    • 在收敛速度与损失值方面: 本文提出的改进算法(Cascade Faster R-CNN)在训练至第47个epoch时收敛,损失值降至0.005。而Faster R-CNN (1)、(2)、(5)的收敛损失值分别为0.21、0.12和0.05。本算法的损失值分别降低了0.205、0.105和0.045,显示出更强的鲁棒性和更优的预测能力。
    • 在检测精度(mAP)方面: 本算法的mAP在训练至第47个epoch时达到最高值94.73%并保持稳定。相比之下,Faster R-CNN (1)的mAP最高为42.10%,(2)为43.30%,(3)为54.96%,(4)为66.94%,(5)为77.40%。本算法的mAP比(1)、(2)、(5)分别提升了52.63%、51.43%和17.33%,提升幅度巨大。与(3)和(4)相比,分别提升了39.77%和27.79%。
    • 在训练速度方面: 本算法的训练速度为0.3650秒/张。虽然并非最快,但与该指标最快的模型(0.3006秒/张)相比,在精度大幅提升的前提下,速度下降在可接受范围内。
  • 与SSD和YOLOv3-tiny算法的性能对比: 研究还将所提算法与单阶段目标检测算法SSD和轻量级网络YOLOv3-tiny进行了比较。结果显示,SSD和YOLOv3-tiny的mAP分别为44.74%和78.50%,训练速度分别为0.0930秒/张和0.2300秒/张, F1分数分别为0.364和0.619。本文算法的mAP(94.73%)和F1分数(0.900)远远高于这两个模型。尽管由于加入了级联模块,其检测速度略慢,但在速度损失不显著的前提下,检测精度获得了决定性的优势。

  • 可视化检测效果: 使用本文改进算法对包含六种缺陷类型的图像进行测试,结果显示,该模型能够准确地定位出不同大小、不同类型的疵点,并以高置信度(大部分在0.89以上)预测出疵点类别。

五、研究结论与价值

本研究得出以下结论:所提出的基于迁移学习与改进Faster R-CNN的织物疵点检测算法,其平均精度均值(mAP)达到了约95%,在六种常见织物疵点的检测任务中,性能和收敛速度均显著优于原始模型及其他主流目标检测算法。该研究的科学价值在于,它系统地证明了结合残差网络、特征金字塔、ROI Align和级联检测器能有效解决深度学习应用于小目标、多尺度工业缺陷检测时的精度瓶颈。其应用价值在于,提供了一种能更好满足工业实际生产需求的自动检测方案,能够有效地降低误检率和漏检率,为替代传统人工检测、实现纺织品的质量控制自动化奠定了坚实的技术基础。

六、研究亮点

本研究的亮点主要体现在以下几个方面:首先,针对织物疵点中大量存在的小目标,创新性地将扩展候选框的RPN、FPN和ROI Align集成到Faster R-CNN框架中,有效解决了对小尺寸疵点检测不准的难题。其次,应用级联模块(Cascade Module),通过逐级优化的策略,实现了检测精度的进一步提升。最后,通过对多种配置的Faster R-CNN以及不同框架的检测算法进行全面、细致的对比分析,以详实的数据充分证明了所提各改进模块的有效性及整体算法的优越性。

七、未来工作展望

本研究的算法目前仍处于测试阶段,作者指出,未来将着重从进一步优化训练速度方面对算法进行改进,以更好地适应实际织物生产过程中的实时性要求。

上述解读依据用户上传的学术文献,如有不准确或可能侵权之处请联系本站站长:admin@fmread.com