基于Swin Transformer与对比学习的皮革缺陷小样本分类方法研究

Research on Small-sample Classification Method of Leather Defects Based on Swin Transformer and Contrastive Learning

  • 摘要: 为了提高深度学习网络的准确度,往往需要利用大量带标签样本进行训练,而样本标记工作耗费大量人力和物力。针对上述问题,提出一种小样本分类方法,将Swin Transformer特征提取模块与对比学习相结合,在一定程度上能够减少训练过程中使用的带标签样本数量,同时保持较高的分类准确度。所建模型分为两个阶段:自监督学习阶段和监督学习阶段。在自监督学习阶段,模型学习同类缺陷之间的共性及不同类别缺陷之间的差异性,经监督学习阶段进行再次调整,以区分皮革的各种缺陷类型。实验结果表明,该方法能够在一定程度上减少带标签训练样本使用量并保持较高的分类准确度,在工业检测领域应用潜力较大。

     

    Abstract: In order to improve the accuracy of deep learning network, it is often necessary to use a large number of labeled samples for training, and sample labeling work consumes a lot of manpower and material resources. In response to the above problems, a small sample classification method is proposed, which combines the feature extraction module of swin transformer with contrastive learning. To a certain extent, it reduces the use of labeled samples during the training process while maintaining a high classification accuracy. The established model is divided into two stages:the self-supervised learning stage and the supervised learning stage. During the self-supervised learning stage, the model learns the commonalities among similar defects and the characteristics among different types of defects. After the learning stage, it is readjustment to distinguish various defect types of leather. The experimental results show that this method can reduce labeled training samples to a certain extent and maintain a high classification accuracy, and has great application potential in the field of industrial inspection.

     

/

返回文章
返回