[發明專利]圖像處理方法、設備和介質

有效

• 申請號: 201911300890.7
• 申請日: 2019-12-17
• 公開（公告）號: CN111027551B
• 公開（公告）日: 2023-07-07
• 發明（設計）人: 管成;郭曉威;余宗橋;孫星;杜俊瓏;彭湃
• 申請（專利權）人: 騰訊科技（深圳）有限公司
• 主分類號: G06V10/25
• 分類號: G06V10/25;G06V10/764;G06V10/82;G06N3/0464;G06N3/096
• 代理公司: 北京市柳沈律師事務所 11105
• 代理人: 王娟;黃玫

1.一種圖像處理方法，包括：

通過第一處理網絡，提取輸入圖像的第一特征圖，并基于所述第一特征圖確定第一置信度圖，其中所述第一處理網絡是完成訓練的網絡，所述第一置信度圖中的每一個位置分別對應于所述輸入圖像中的相應位置，且包括目標框信息以及與所述目標框對應的類別置信度，所述目標框信息為指示包圍所述輸入圖像中的相應位置處檢測到的對象的框的相關信息，且所述類別置信度為指示所述對象屬于特定類別的置信度的信息；

通過對所述第一置信度圖執行調整處理，得到參考置信度圖；

通過第二處理網絡，提取輸入圖像的第二特征圖，并基于所述第二特征圖確定第二置信度圖；

至少基于所述第二置信度圖與所述參考置信度圖之間的損失函數，訓練所述第二處理網絡；以及

基于完成訓練的第二處理網絡，對向其輸入的圖像執行目標檢測處理，

其中，第一處理網絡的參數量大于第二處理網絡的參數量。

2.根據權利要求1所述的方法，其中，所述第一置信度圖中的每一個位置對應一個置信度向量，且所述置信度向量包括目標框信息以及與所述目標框對應的類別置信度。

3.根據權利要求1所述的方法，其中通過對所述第一置信度圖執行調整處理，得到參考置信度圖進一步包括：

針對所述第一置信度圖中的每一個位置，執行以下處理：

確定在所述第一置信度圖中的該位置以及相鄰位置包括的類別置信度；

在所述第一置信度圖中的該位置以及相鄰位置包括的目標框中，確定屬于同一個目標框的多個目標框；

對所述多個目標框的類別置信度進行排序；以及

將最大的類別置信度以及對應的目標框信息作為第一置信度圖中該位置包括的類別置信度以及對應的目標框信息，并且當該位置還包括其他目標框信息時，將其他目標框信息所對應的類別置信度置為0。

4.根據權利要求3所述的方法，其中確定在所述第一置信度圖中的該位置以及相鄰位置包括的類別置信度進一步包括：

基于相鄰位置與該位置之間的距離，對相鄰位置的類別置信度進行加權，以作為相鄰位置的類別置信度。

5.根據權利要求3所述的方法，其中在確定在所述第一置信度圖中的該位置以及相鄰位置包括的類別置信度之后，進一步包括：

過濾各類別置信度，以將低于第一預定閾值的類別置信度置為0。

6.根據權利要求3所述的方法，其中確定屬于同一個目標框的多個目標框進一步包括：

確定兩個目標框之間的重疊程度；

當所述重疊程度大于第二預定閾值時，確定所述兩個目標框屬于同一個目標框。

7.根據權利要求1所述的方法，其中所述輸入圖像為一訓練數據集中的圖像，且對應于一標注的目標檢測結果，所述方法進一步包括：

基于所述標注的目標檢測結果，生成標注置信度圖；

其中至少基于所述第二置信度圖與所述參考置信度圖之間的損失函數，訓練所述第二處理網絡進一步包括：

基于所述第二置信度圖與所述參考置信度圖之間的損失函數以及所述第二置信度圖與所述標注置信度圖之間的損失函數，訓練所述第二處理網絡。