基于融合多头注意力重建网络的无监督丝织物瑕疵检测方法

本发明涉及一种基于融合多头注意力重建网络的无监督丝织物瑕疵检测方法，属于织物瑕疵检测技术领域。所述方法，提出一种基于融合多头注意力重建网络的丝织物瑕疵检测框架，该框架由Transformer预训练编码器、带有MeanDropout的瓶颈层和融合注意力解码器组成。具体来说，首先使用预训练视觉Transformer(ViT)作为特征编码器，以提取丝织物图像的判别特征作为重建对象；然后，将提取的特征输入带有MeanDropout的瓶颈层，从而减少模型对重复特征的依赖；此外，提出一种融合多头注意力机制(HMA)和LLamaMLP相结合的解码器以指导注意力头动态选择适当的细微特征；最后，由改良后的解码器将多层特征组合为一个整体进行松散重建，即不强制层对层和逐点重建，从而放宽重构约束。