[发明专利]一种基于Swin-Transformer与NAS-FPN的织物疵点检测方法

摘要

本发明涉及一种基于Swin‑Transformer与NAS‑FPN的织物疵点检测方法，主要有三个阶段：第一阶段，建立数据集，收集带有疵点的织物图像，人工标注疵点的位置和类型，并将数据集划分为训练集和测试集两个部分；第二阶段：训练模型，在训练集上，通过Swin‑Transformer模块提取输入图像特征，然后使用NAS‑FPN模块进行特征融合，将获得的特征通过RPN模块得到粗略的疵点位置框；第三阶段，通过对特征的处理得到最终织物疵点的位置和类型。本发明将Swin‑Transformer与NAS‑FPN相结合搭建的目标检测模型比通用的目标检测网络检测结果更加准确，特别是对于织物疵点更有针对性。