基于工业大模型的零部件识别方法及装置

本发明涉及零部件识别技术领域，公开了一种基于工业大模型的零部件识别方法及装置，该方法包括：对工业零部件的原始图像进行图像块划分和嵌入变换，得到第一Token序列，并对第一Token序列进行二维离散余弦变换，得到第二Token序列；将第二Token序列与类别原型记忆库中选取多个原型特征向量进行融合，得到记忆引导向量；基于记忆引导向量对第二Token序列进行门控调制和残差连接处理，得到第三Token序列；将第三Token序列执行迭代细化，得到第四Token序列，并对第四Token序列进行全局池化和分类预测，得到工业零部件的类别识别结果，本发明增强了零部件特征的判别性，有效解决了稀有零部件小样本识别困难的技术问题。