Segmentación de cuerpos de agua en imágenes de teledetección basada en UNet mejorado con atención de fusión multiescala

SHI Tian-tian ,  

GUO Zhong-hua ,  

YAN Xiang ,  

WEI Shi-qin ,  

摘要

Para la tarea de segmentación de cuerpos de agua en imágenes de teledetección, se propuso una red UNet mejorada con un módulo de atención de fusión multiescala — A-MSFAM-UNet. Este método realiza una segmentación de cuerpos de agua de imágenes de alta resolución de extremo a extremo en la tarea de segmentación de agua de imágenes GF-2. Primero, para abordar el problema de insensibilidad a la información local causada por la operación de pooling global en módulos de atención anteriores, se diseñó un módulo de atención de fusión multiescala (MSFAM), que utiliza convolución puntual para fusionar información global de los canales y convolución separable en profundidad para compensar la pérdida de información causada por el pooling global. MSFAM se utiliza en la parte de fusión de características después de las conexiones de salto de UNet para redistribuir los pesos de los puntos de características y mejorar la eficiencia de la fusión de características, fortaleciendo la capacidad de la red para obtener información de diferentes escalas. En segundo lugar, se utiliza convolución dilatada en la red principal VGG16 para expandir el campo receptivo y agregar información global sin perder resolución. Los resultados muestran que A-MSFAM-UNet supera a otros UNet mejorados con atención de canal (SENet, ECANet), alcanzando un índice de intersección sobre unión promedio (MIoU), precisión media por píxel (MPA) y exactitud (Acc) de 96.02%, 97.98% y 99.26% respectivamente en el conjunto de datos de segmentación de cuerpos de agua GF-2.

关键词

imágenes de teledetección;módulo de atención;convolución separable en profundidad;fusión de características;convolución dilatada

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website