结合U-Net和分割模型的遥感影像建筑物提取方法

本发明公开结合U‑Net和分割模型的遥感影像建筑物提取方法，包括：获取遥感影像，通过U‑Net处理遥感影像，确定各建筑物的第一掩膜；根据各建筑物的第一掩膜，确定各建筑物的点提示；将各点提示输入SAM分割，确定各建筑物的第二掩膜；根据各第一掩膜和各第二掩膜，确定各建筑物的差分掩膜；判断差分掩膜的面积是否符合预设面积；当符合预设面积时，第二掩膜为最终掩膜；当不符合预设面积时，将差分掩膜作为第一掩膜，执行根据各建筑物的第一掩膜，确定各建筑物的点提示的步骤。本发明在不增加运行时间的前提下有效提升了建筑物分割的完整性和降低了误检率，有效的避免了掩膜识别不全和细节缺失的问题。