基于迁移学习的红外图像识别方法及存储介质

摘要

本发明公开了一种基于迁移学习的红外图像识别方法及存储介质，方法包括：根据红外样本图像，获取正样本图像块和负样本图像块，并划分得到第一训练集、第二训练集和测试集；构建红外图像处理模型，包括特征提取模块、标签预测模块和域分类器；根据训练集，分别对多个红外图像处理模型中的特征提取模块和域分类器进行训练；根据测试集，分别对训练后的多个红外图像处理模型进行评估，选取最优红外图像处理模型；根据第一训练集和测试集，对最优红外图像处理模型中的标签预测模块进行训练；将待识别图像块输入训练后的最优红外图像处理模型，并将标签预测模块的输出结果作为识别结果。本发明可提高红外图像的识别准确率和识别速度。