[發明專利]水體提取方法、裝置、電子設備及存儲介質

在審

• 申請號: 202110004595.8
• 申請日: 2021-01-04
• 公開（公告）號: CN112668530A
• 公開（公告）日: 2021-04-16
• 發明（設計）人: 于強;郭曉瑞;張寶才;彭乘風
• 申請（專利權）人: 北京簡巨科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/34;G06K9/40;G06K9/46;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京風雅頌專利代理有限公司 11403
• 代理人: 郭曼

1.一種水體提取方法，包括：

獲取包含水體的遙感影像；

提取所述遙感影像的光譜特征或極化特征；

將所述光譜特征或極化特征作為水體提取模型的輸入特征；以及

基于水體提取模型，根據所述輸入特征，識別出所述遙感影像中的水體；其中，所述水體提取模型是用于基于輸入特征識別出遙感影像中水體的基于殘差網絡的深度學習分割網絡U-net模型。

2.根據權利要求1所述的方法，其中，所述遙感影像包括光學遙感影像；所述光譜特征包括綠、紅、藍波段數據和/或近紅外波段數據。

3.根據權利要求1所述的方法，其中，所述遙感影像包括微波遙感影像；所述極化特征包括至少兩種極化方式的極化數據。

4.根據權利要求1所述的方法，進一步包括：

獲取所述遙感影像各像素點的水體指數；以及

將獲取的水體指數作為所述水體提取模型的輸入特征之一。

5.根據權利要求1至4中任一項所述的方法，進一步包括：對所述輸入特征進行輻射定標處理。

6.根據權利要求1所述的方法，其中，所述基于殘差網絡的U-net模型包括特征提取部分以及上采樣部分；其中，

所述特征提取部分包括至少一個特征提取層和一個中間層；其中，每個特征提取層分別包括兩個殘差網絡單元層和一個池化層；所述中間層包括兩個殘差網絡單元層和一個反卷積層；

所述上采樣部分包括至少一個上采樣層和一個輸出層；其中，每個上采樣層分別包括兩個殘差網絡單元層和一個反卷積層；所述輸出層包括兩個殘差網絡單元層和一個卷積層。

7.根據權利要求6所述的方法，其中，所述基于殘差網絡的U-net模型進一步包括：位于所述兩個殘差網絡單元層之間的批量歸一化BN層。

8.一種水體提取裝置，包括：

遙感影像獲取模塊，用于獲取包含水體的遙感影像；

特征提取模塊，用于提取所述遙感影像的光譜特征或極化特征，并將所述光譜特征或極化特征作為水體提取模型的輸入特征；

識別模塊，用于基于水體提取模型，根據所述輸入特征，識別出所述影像中的水體；其中，所述水體提取模型是用于基于輸入特征識別出影像中水體的基于殘差網絡的深度學習分割網絡U-net模型。

9.一種電子設備，包括存儲器、處理器及存儲在存儲器上并可在處理器上運行的計算機程序，其中，所述處理器執行所述程序時實現如權利要求1至4以及6至7中任意一項所述的水體提取方法。

10.一種非暫態計算機可讀存儲介質，其中，所述非暫態計算機可讀存儲介質存儲計算機指令，所述計算機指令用于使所述計算機執行權利要求1至4以及6至7中任意一項所述的水體提取方法。