[发明专利]一种基于图像检索增强的航拍图像分类方法及装置

摘要

本发明公开了一种基于图像检索增强的航拍图像分类方法及装置，方法包括数据集构建、模型构建、模型训练以及分类步骤，在主干网络中，特征嵌入经过全连接层的前向传播得到基础分类概率；在检索分支中，特征嵌入在检索表中进行得到相应文本信息，后通过文本编码器的前向传播得到检索的分类概率；将两个分类概率进行归一化和求和，经过角度惩罚模块得到最终的分类概率；然后计算角度惩罚损失，反向传播更新网络参数，最终得到分类模型，最后利用分类模型进行测试。本发明能够利用检索来增加分类网络预测的分类概率，增强分类性能，能够利用角度惩罚损失，增加惩罚裕度，直接优化测地距离弧度，使得每个类具有更高的可区分性，提高了分类的准确性。