来源:期刊VIP网 时间:
作者:王民水;王明常;王婧瑜;刘子维;
单位:吉林大学地球探测科学与技术学院;吉林省农业科学院;
摘要:针对城市遥感图像各种地物分布不均衡、分类精度较低的问题,提出融合并行注意力与权重平衡算法的遥感图像分类方法。该方法在DeepLabV3+和ResNet50创建的语义分割网络基础上,采用并行组合方式,融入通道注意力和空间注意力算法,提高网络的特征提取能力;针对遥感图像地物类别占比不均衡问题,引入地物类别权重平衡算法,提高小类别地物的分类精度。为了验证网络模型的分类效果,利用Vaihingen数据集和Postdam数据集进行实验。实验结果表明:融合注意力机制和权重平衡算法的分类网络在Vaihingen数据集中测试数据的像素精度、平均交并比、平均F_1值分别为96.66%、90.35%、96.66%,在Postdam数据集中测试数据的像素精度、平均交并比、平均F_1值分别为95.74%、81.47%、91.82%;从分类细节看,增加注意力机制和权重平衡算法对占比较少的汽车识别精度有显著提高,在Vaihingen数据集中汽车的像素精度提高了26.44%,在Postdam数据集中汽车的像素精度提高了21.84%,取得了较好的分类效果。
关键词:注意力机制;;权重平衡算法;;DeepLabV3+网络;;遥感图像;;地物分类
基金资助:国家自然科学基金项目(42171407)~~