一种基于尺度注意力网络的遥感图像场景分类方法

作者：边小勇; 费雄君; 穆楠; 张晓龙; 邓春华

来源：2019-07-09, 中国, ZL201910614535.0.

本发明公开了一种基于尺度注意力网络的遥感图像场景分类方法,首先,将场景数据集按比例随机划分为训练集和测试集；接着,对数据集进行预处理,包括图像缩放和归一化；同时,将数据集输入到注意力模块中进行显著性检测,产生注意力图；然后,利用预训练模型初始化尺度注意力网络参数,并使用训练集和注意力图微调尺度注意力网络,保存训练好的网络模型；最后,使用微调后的尺度注意力网络预测待分类图像场景的类别。该基于尺度注意力网络的遥感图像场景分类方法,通过多次利用多尺度注意力图给特征图加权,又通过提取多尺度图像特征并融合,产生了判别力增强的特征表示,在遥感图像场景分类上取得了更好的效果。

• 单位
  武汉科技大学