一种基于行列自注意力全卷积神经网络的遥感图像鱼塘提取方法

摘要

本发明公开了一种基于自注意力全卷积神经网络的遥感图像鱼塘提取方法，基本步骤为：利用遥感图像计算NDWI，并利用NDWI进行初步分类，然后制作遥感图像样本集和标定样本集；接下来构建自注意力全卷积神经网络并训练模型，接着用训练后模型对遥感图像进行预测，得到遥感图像各像元类别概率分布；最后将各像元类别概率分布与NDWI分类的结果进行融合，得到分类结果。本发明在使用全卷积神经网络的基础上，设计了行列双向GRU自注意力模型，对遥感图像中的鱼塘提取具有准确度高、边缘完整的特点，可用于鱼塘自动化提取及变化检测等应用。