一种基于轻量级光谱注意力网络的铝合金牌号识别方法

本发明属于光谱技术和深度学习技术领域，提供一种基于轻量级光谱注意力网络的铝合金牌号识别方法，用以提高铝合金牌号识别的准确性和鲁棒性。本发明首先，通过格拉姆场变换将一维XRF光谱转换为二维谱图，用以丰富空间信息并放大光谱细节差异；然后，提出轻量级光谱注意力模块，该模块通过简单的自适应计算来减少计算量、关注谱图的重要信息并抑制冗余信息；最后，在轻量级空间注意力模块的基础上结合ShuffleNet结构设计了轻量级空间自关注网络，实现铝合金牌号的准确识别，有效提高铝合金牌号识别的准确性和鲁棒性；并且，本发明中铝合金牌号识别方法作为一种端到端的深度学习模型，无需对样本进行复杂预处理即可准确识别出合金的牌号。