[發明專利]光伏板圖像分割方法及其模型的訓練方法、裝置

在審

• 申請號: 202310618102.9
• 申請日: 2023-05-29
• 公開（公告）號: CN116630624A
• 公開（公告）日: 2023-08-22
• 發明（設計）人: 徐胤;毛晴;黃夢珂;楊家榮;韓少恒
• 申請（專利權）人: 上海電氣集團股份有限公司
• 主分類號: G06V10/26
• 分類號: G06V10/26;G06V10/774;G06V10/82;G06N3/0895
• 代理公司: 上海弼興律師事務所 31283
• 代理人: 羅朗;林嵩

1.一種光伏板圖像分割模型的訓練方法，其特征在于，所述訓練方法包括：

獲取訓練樣本集，所述訓練樣本集包括具有真實標簽的多個第一樣本圖像以及無標簽的多個第二樣本圖像，所述第一樣本圖像和所述第二樣本圖像均包含光伏板，所述第一樣本圖像的數量少于所述第二樣本圖像的數量；

將所述第一樣本圖像輸入第一模型進行光伏板分割處理，輸出所述第一樣本圖像的第一圖像分割結果；

將所述第二樣本圖像輸入第二模型進行光伏板分割處理，輸出所述第二樣本圖像的置信度和第二圖像分割結果，并根據所述第二圖像分割結果得到所述第二樣本圖像的偽標簽；

根據所述置信度從所述第二樣本圖像中選擇第三樣本圖像，并從所述偽標簽中選擇與所述第三樣本圖像對應的目標偽標簽；

將所述第三樣本圖像輸入所述第一模型進行光伏板分割處理，輸出所述第三樣本圖像的第三圖像分割結果；

根據所述真實標簽和所述第一圖像分割結果計算的第一損失以及所述目標偽標簽和所述第三圖像分割結果計算的第二損失更新所述第一模型的模型參數，并基于所述第一模型的模型參數更新所述第二模型的模型參數，直至滿足收斂條件，將收斂的第一模型作為所述光伏板圖像分割模型。

2.根據權利要求1所述的光伏板圖像分割模型的訓練方法，其特征在于，所述根據所述置信度從所述第二樣本圖像中選擇第三樣本圖像的步驟包括：

選擇所述置信度大于置信度閾值的所述第二樣本圖像作為所述第三樣本圖像；

或，

選擇所述置信度排序靠前的目標數量的所述第二樣本圖像作為所述第三樣本圖像。

3.根據權利要求1所述的光伏板圖像分割模型的訓練方法，其特征在于，所述根據所述真實標簽和所述第一圖像分割結果計算的第一損失以及所述目標偽標簽和所述第三圖像分割結果計算的第二損失更新所述第一模型的模型參數的步驟包括：

確定所述第一損失與所述第二損失的加權和，得到目標損失；其中，所述第一損失的權重不小于所述第二損失的權重；

根據所述目標損失更新所述第一模型的模型參數。

4.根據權利要求3所述的光伏板圖像分割模型的訓練方法，其特征在于，所述訓練方法還包括：

在訓練過程中逐漸增加所述第二損失的權重。

5.根據權利要求1所述的光伏板圖像分割模型的訓練方法，其特征在于，所述獲取訓練樣本集的步驟之后還包括：

對所述第一樣本圖像和所述第二樣本圖像進行增強處理，所述增強處理包括隨機裁剪、對比度變換、翻轉和/或隨機拼接。

6.一種光伏板圖像分割方法，其特征在于，所述光伏板圖像分割方法包括：

獲取待分割圖像，所述待分割圖像包含光伏板；

將所述待分割圖像輸入光伏板圖像分割模型進行光伏板分割處理，得到所述待分割圖像的光伏板圖像分割結果；

其中，所述光伏板圖像分割模型通過如權利要求1-5任一項所述的光伏板圖像分割模型的訓練方法得到。