基于融合注意力机制的小样本遥感场景分类方法 |
| |
引用本文: | 李子茂,于舒,郑禄,帖军,秦锦添.基于融合注意力机制的小样本遥感场景分类方法[J].国外电子测量技术,2023(7):59-67. |
| |
作者姓名: | 李子茂 于舒 郑禄 帖军 秦锦添 |
| |
作者单位: | 1. 中南民族大学计算机科学学院;3. 农业区块链与智能管理湖北省工程研究中心 |
| |
摘 要: | 针对遥感场景图像样本获取困难,数据量受限以及遥感图像目标对象和背景高度混杂的问题,提出一种基于融合注意力机制的小样本遥感场景分类方法。该方法采用RepVGG作为基准模型,并利用ECANet网络的ECA通道注意力机制改进RepVGG网络的RepVGGBlock模块,使得网络有效过滤无用信息并聚焦于关键场景区域,从而增强模型的特征判别能力,并确保在不增加模型参数的情况下提高分类准确率;同时通过随机数据增强方法在线增强训练数据,在不占用额外内存的情况下增加模型训练数据量,使得训练数据更多样化,提高模型的泛化能力。在UC Merced LandUse数据集上分类平均准确率为94.52%,相较于ResNet50、RepVGG-B1-SE网络,准确率分别提高4.52%和2.93%。实验结果表明,该方法能有效聚焦关键场景区域并提升小样本遥感场景分类的准确率,对实现遥感影像快速分类具有一定的参考意义。
|
关 键 词: | 遥感场景分类 注意力机制 RepVGG网络 小样本 ECANet |
|
|