基于多尺度融合注意力改进UNet的遥感图像水体分割 |
| |
引用本文: | 石甜甜,郭中华,闫翔,魏士钦.基于多尺度融合注意力改进UNet的遥感图像水体分割[J].液晶与显示,2023(3):397-408. |
| |
作者姓名: | 石甜甜 郭中华 闫翔 魏士钦 |
| |
作者单位: | 1. 宁夏大学物理与电子电气工程学院;2. 宁夏大学沙漠信息智能感知重点实验室 |
| |
基金项目: | 宁夏自然科学基金(No.2020AAC03026); |
| |
摘 要: | 针对遥感图像水体分割任务,提出了一种多尺度融合注意力模块改进的UNet网络——A-MSFAM-UNet,该方法在GF-2遥感图像水体分割任务中实现了端到端高分辨率遥感图像水体分割。首先,针对以往注意力模块全局池化操作带来的局部信息不敏感问题,设计了一种多尺度融合注意力模块(MSFAM),该模块使用点卷积融合通道全局信息、深度可分离卷积弥补全局池化造成的信息丢失。MSFAM用于UNet跳跃连接后的特征融合部分重新分配特征点权重以提高特征融合效率,增强网络获取不同尺度信息的能力。其次,空洞卷积用于VGG16主干网络扩展感受野,在不损失分辨率的情况下聚合全局信息。结果表明,A-MSFAM-UNet优于其他通道注意力(SENet、ECANet)改进的UNet,在GF-2水体分割数据集上平均交并比(MIoU)、平均像素精度(MPA)和准确率(Acc)分别达到了96.02%、97.98%和99.26%。
|
关 键 词: | 遥感图像 注意力模块 深度可分离卷积 特征融合 空洞卷积 |
|