[發明專利]目標識別網絡的訓練方法、裝置和系統

在審

• 申請號: 202210579955.1
• 申請日: 2022-05-25
• 公開（公告）號: CN114821242A
• 公開（公告）日: 2022-07-29
• 發明（設計）人: 金國強;孫明珊;李韡;暴天鵬;吳立威
• 申請（專利權）人: 上海商湯智能科技有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/10;G06K9/62;G06V10/28;G06V10/32;G06V10/764;G06V10/82;G06N3/04;G06N5/04;G06T5/50;G06T7/11;G06T7/194;G06T7/70
• 代理公司: 北京博思佳知識產權代理有限公司 11415
• 代理人: 董曉盈

1.一種目標識別網絡的訓練方法，其特征在于，所述方法包括：

獲取接觸網圖像；

將所述接觸網圖像輸入至預先訓練的熱力圖生成網絡，得到所述接觸網圖像的熱力圖；

根據所述熱力圖得到至少一個對象的預選框；

根據所述接觸網圖像和所述預選框，訓練目標識別網絡。

2.根據權利要求1所述的方法，其特征在于，所述將所述接觸網圖像輸入至預先訓練的熱力圖生成網絡，得到所述接觸網圖像的熱力圖，包括：

將所述接觸網圖像輸入至預先訓練的熱力圖生成網絡；

通過所述熱力圖生成網絡中的自注意力機制，得到多張子熱力圖；

通過對所述多張子熱力圖進行加權融合，得到所述接觸網圖像的熱力圖。

3.根據權利要求1所述的方法，其特征在于，所述根據所述熱力圖得到至少一個對象的預選框，包括：

對所述熱力圖中各像素點的像素值進行二值化處理；

基于二值化處理結果，得到至少一個連通區域；

根據所述連通區域的矩形框，得到對應對象的預選框。

4.根據權利要求1所述的方法，其特征在于，所述根據所述熱力圖得到至少一個對象的預選框，包括：

根據所述接觸網圖像的尺寸，調整所述熱力圖的尺寸，得到調整后熱力圖；

根據所述調整后熱力圖得到至少一個對象的預選框。

5.根據權利要求1所述的方法，其特征在于，所述根據所述接觸網圖像和所述預選框，訓練目標識別網絡，包括：

將預選框映射至所述接觸網圖像，獲取映射結果中所述預選框對應的圖像區域；

獲取所述圖像區域的標注特征信息和第一標注類別信息，所述第一標注類別信息指示所述預選框中的對象是前景；

利用第一樣本圖像訓練目標識別網絡；其中，所述第一樣本圖像通過所述圖像區域、對應的所述標注特征信息和所述第一標注類別信息構建。

6.根據權利要求5所述的方法，其特征在于，所述利用所述第一樣本圖像訓練目標識別網絡，包括：

將所述第一樣本圖像輸入目標識別網絡，得到所述第一樣本圖像中的第一檢測框，以及所述第一檢測框對應圖像區域的第一預測類別信息，所述第一預測類別信息指示所述第一檢測框中對象是前景或背景；

獲取所述第一檢測框對應圖像區域的預測特征信息；

根據所述第一樣本圖像中預選框與所述第一檢測框之間的位置差異，所述第一標注類別信息與所述第一預測類別信息之間的類別差異，以及所述標注特征信息與所述預測特征信息之間的特征差異，調整所述目標識別網路的網絡參數。

7.根據權利要求1-6中任意一項所述的方法，其特征在于，所述方法還包括：

獲取包括目標對象的標注信息的接觸網圖像作為第二樣本圖像，其中，所述標注信息包括標注框以及所述標注框對應的第二標注類別信息，所述第二標注類別信息指示所述標注框內目標對象的類別；

基于所述第二樣本圖像對所述目標識別網絡進行訓練。

8.根據權利要求7所述的方法，其特征在于，所述基于所述第二樣本圖像對所述目標識別網絡進行訓練，包括：

將所述第二樣本圖像輸入目標識別網絡進行識別，得到所述第二樣本圖像的識別結果，所述識別結果包括第二檢測框以及所述第二檢測框對應圖像區域的第二預測類別信息；

根據所述標注框與所述第二檢測框之間的差異，以及所述第二標注類別信息與所述第二預測類別信息的差異，調整所述目標識別網絡的網絡參數。

9.根據權利要求7所述的方法，其特征在于，所述目標對象至少包括下列中的至少一個：絕緣子、驅鳥罩、接觸線、吊弦和承力索。

10.一種目標識別方法，其特征在于，所述方法包括：

獲取待識別圖像；

將所述待識別圖像輸入目標識別網絡，輸出目標識別結果，所述目標識別網絡基于權利要求1-9中任意一項權利要求所述方法得到。