一种基于ViT的航空图像识别方法
摘要:
本发明公开了一种基于ViT的航空图像识别方法,包括步骤:S1、采集航空图像数据集,构造训练集,验证集和测试集;S2、扩充训练集的数据量;S3、构建基于ViT的航空图像识别模型;S4、将扩充后的训练集输入到识别模型中,对图像相对应的标签进行区分性标签平滑,同时采用交叉熵损失函数和区分性对比损失函数对模型进行训练,通过反向传播算法更新识别模型,遴选最优的航空图像识别模型;S5、通过测试集测试模型的识别性能。本发明通过将图像对应的标签进行区分性标签平滑处理,同时采用交叉熵损失函数和区分性对比损失函数监督模型的训练过程,得到具有更强特征学习能力的基于ViT的航空图像识别模型,具有识别率高,可拓展性强等优点。
公开/授权文献
0/0