基于CNN-Transformer的遥感图像水体提取方法、装置、电子设备及介质

    公开(公告)号:CN117036805A

    公开(公告)日:2023-11-10

    申请号:CN202311003802.3

    申请日:2023-08-10

    Abstract: 本公开实施例提供了一种基于CNN‑Transformer的遥感图像水体提取方法、装置、电子设备及介质,方法包括:将待提取的遥感图像输入至预先训练的基于CNN‑Transformer的深度学习模型;基于CNN的多尺度特征提取技术对所述遥感图像进行特征编码,获得所述遥感图像的多尺度编码特征;基于Mobile ViT的Unfold‑Transformer‑Fold结构对所述多尺度编码特征进行处理,获得所述遥感图像的全局编码特征;对所述全局编码特征进行解码得到解码特征,基于协调注意力机制,对相同尺寸的多尺度编码特征和解码特征两两进行融合,得到融合特征;对所述融合特征进行解码,得到遥感图像中的水体区域。本实施例能够以较小的参数量,有效利用网络深层的上下文语义信息,实现多层次、多尺度水体信息的快速、精确提取。

Patent Agency Ranking