买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:武汉理工大学
摘要:本发明公开了一种基于ViT的航空图像识别方法,包括步骤:S1、采集航空图像数据集,构造训练集,验证集和测试集;S2、扩充训练集的数据量;S3、构建基于ViT的航空图像识别模型;S4、将扩充后的训练集输入到识别模型中,对图像相对应的标签进行区分性标签平滑,同时采用交叉熵损失函数和区分性对比损失函数对模型进行训练,通过反向传播算法更新识别模型,遴选最优的航空图像识别模型;S5、通过测试集测试模型的识别性能。本发明通过将图像对应的标签进行区分性标签平滑处理,同时采用交叉熵损失函数和区分性对比损失函数监督模型的训练过程,得到具有更强特征学习能力的基于ViT的航空图像识别模型,具有识别率高,可拓展性强等优点。
主权项:1.一种基于ViT的航空图像识别方法,其特征在于:所述方法包括如下步骤:S1采集航空图像数据集,得到所需原始航空图像xi及其对应的类别标签yi,按比例数量划分训练集、验证集和测试集,分别用于后续对模型进行训练、验证和评估,其中训练集记为B为训练集的图像数目;S2将所述训练集图像进行在线数据增强,使得训练集中每张图像都生成M张不同的增强图像,训练集被扩充后的图像数量为B*M,记为S3构建基于ViT的航空图像识别模型;S4将所述训练集的图像,输入所述基于ViT的航空图像识别模型,对图像相对应的标签进行区分性标签平滑,同时采用交叉熵损失函数和区分性对比损失函数对模型进行训练,通过反向传播算法更新识别模型,并利用步骤S1中的验证集遴选最优的航空图像识别模型;S5使用步骤S1的测试集测试所述航空图像识别模型的识别性能,得到最终的模型识别正确率,当模型识别正确率达到设定阈值时,将待识别图像输入航空图像识别模型进行识别;否则返回步骤S3直至模型识别正确率达到设定阈值。
全文数据:
权利要求:
百度查询: 武汉理工大学 一种基于ViT的航空图像识别方法
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。