[发明专利]基于注意力机制的图像识别方法、系统、设备和存储介质

摘要

本申请涉及一种基于注意力机制的图像识别方法、系统、设备和存储介质。该方法包括：获取待识别图像的小样本数据集，将小样本数据集经过多头注意力模型进行图像的特征向量处理；将处理后得到的特征向量矩阵经过高斯混合模型进行预处理，使得特征向量矩阵具备高斯分布；将分布一致的特征向量矩阵经过最优运输原理算法和距离度量算法进行聚类计算，得到聚类特征信息，根据聚类特征信息获取图像识别结果。本发明以层级化的多头注意力特征提取来更全面地提取图像中的关键信息，使得图像识别的广度和范围显著提高，同时对距离算法进行了改进，结合最优运输原理的思想，实现了计算非精确距离重心，避免了过拟合问题，进一步提高了图像识别准确度。