[發明專利]云去除方法、裝置、計算機設備和存儲介質

在審

• 申請號: 202210410940.2
• 申請日: 2022-04-19
• 公開（公告）號: CN114742733A
• 公開（公告）日: 2022-07-12
• 發明（設計）人: 張佳穎
• 申請（專利權）人: 中國工商銀行股份有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06N3/04;G06N3/08
• 代理公司: 華進聯合專利商標代理有限公司 44224
• 代理人: 景懷宇

1.一種云去除方法，其特征在于，所述方法包括：

獲取待處理圖像，所述待處理圖像是存在云層覆蓋的遙感圖像；

基于所述預先訓練的云去除模型，對所述待處理圖像進行第一類型卷積計算，得到第一類型特征圖，以及對所述待處理圖像進行第二類型卷積計算，得到第二類型特征圖；

根據預設的混合注意力算法，對所述第一類型特征圖進行處理，得到增強特征圖，根據預設的密集殘差算法，對所述第二類型特征圖進行密集殘差計算，得到密集殘差；

對所述增強特征圖以及所述密集殘差進行融合處理，得到第一融合特征；

根據所述第一融合特征進行圖像重建，得到所述待處理圖像對應的目標無云圖像。

2.根據權利要求1所述的方法，其特征在于，所述獲取待處理圖像，包括：

采集初始圖像，識別所述初始圖像中的云層區域，得到所述初始圖像對應的第一有云區域圖像以及第一無云區域圖像，并將所述第一有云區域圖像作為所述待處理圖像；

所述方法還包括：

將所述第一無云區域圖像以及所述目標無云圖像進行疊加，得到所述初始圖像對應的無云圖像。

3.根據權利要求1所述的方法，其特征在于，所述對所述待處理圖像進行第一類型卷積計算，得到第一類型特征圖，包括：

根據多個不同的預設尺度，分別對所述待處理圖像進行卷積計算，得到所述待處理圖像對應的多個不同維度的初始特征圖，所述維度與所述預設尺度一一對應；

對兩兩初始特征圖進行融合拼接處理，得到所述待處理圖像更新后的初始特征圖；

在所述更新后的初始特征圖不滿足預設單一條件的情況下，重新執行所述對兩兩初始特征圖進行融合拼接處理，得到所述待處理圖像更新后的初始特征圖的步驟，直至所述更新后的初始特征圖滿足預設單一條件，并將滿足所述預設單一條件的初始特征圖作為第一類型特征圖。

4.根據權利要求1所述的方法，其特征在于，所述根據預設的混合注意力算法，對所述第一類型特征圖進行處理，得到增強特征圖，包括：

對所述第一類型特征圖進行殘差計算，得到目標殘差特征；

根據預設的混合注意力算法，對所述目標殘差特征進行增強處理，得到增強特征圖。

5.根據權利要求1所述的方法，其特征在于，所述根據所述第一融合特征進行圖像重建，得到所述待處理圖像對應的目標無云圖像，包括：

對所述第一融合特征進行目標次數的上采樣處理，得到恢復像素數據；

對所述恢復像素數據進行所述第二類型卷積計算，得到所述待處理圖像對應的目標無云圖像。

6.根據權利要求4所述的方法，其特征在于，所述云去除模型包括殘差計算模塊，所述殘差計算模塊包括融合單元以及多個殘差計算單元；

所述對所述第一類型特征圖進行殘差計算，得到目標殘差特征，包括：

通過所述殘差計算模塊中包括的所述融合單元以及多個殘差計算單元，對所述第一類型特征圖進行殘差計算，得到目標殘差特征；

其中，各個所述殘差計算單元的輸入端與輸出端依次相接，且第一個所述殘差計算單元的輸出端與所述多個殘差計算單元中的目標殘差計算單元的輸出端進行融合后，與所述目標殘差計算單元的下一個所述殘差計算單元的輸入端連接，第一個所述殘差計算單元的輸出端與最后一個所述殘差計算單元的輸出端進行融合后，與所述融合單元的輸入端連接。

7.根據權利要求4所述的方法，其特征在于，所述預設的混合注意力算法包括預設池化算法以及預設通道算法；所述根據預設的混合注意力算法，對所述目標殘差特征進行增強處理，得到增強特征圖，包括：

根據預設池化算法，對所述目標殘差進行池化計算，得到初始空間特征；

根據預設通道算法，對所述目標殘差進行通道計算，得到初始通道特征；

根據所述初始空間特征以及所述初始通道特征，確定增強特征圖。