[發明專利]樣本圖像的處理方法、裝置、設備及存儲介質

在審

• 申請號: 202011543738.4
• 申請日: 2020-12-24
• 公開（公告）號: CN112733886A
• 公開（公告）日: 2021-04-30
• 發明（設計）人: 聶泳忠;楊素偉
• 申請（專利權）人: 西人馬帝言（北京）科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06N3/04;G06N3/08;G06T5/00
• 代理公司: 北京東方億思知識產權代理有限責任公司 11258
• 代理人: 趙秀芹

1.一種樣本圖像的處理方法，其特征在于，所述方法包括：

獲取樣本圖像的多個第一目標框，其中，所述第一目標框包括第一目標圖像，所述第一目標圖像包括通道特征和空間特征；

使用預設的神經網絡中包括的通道特征計算模塊和空間特征計算模塊對每個所述第一目標框包括的第一目標圖像進行計算，得到特征增強后的目標圖像集；

將所述特征增強后的目標圖像集添加至所述樣本圖像中，得到增強后的圖像。

2.根據權利要求1所述的方法，其特征在于，所述使用預設的神經網絡中包括的通道特征計算模塊和空間特征計算模塊對所述每個所述第一目標框包括的第一目標圖像進行計算，得到特征增強后的目標圖像集，包括：

使用所述通道特征計算模塊對每個所述第一目標圖像的每個通道特征進行計算，得到每個所述第一目標圖像的通道權重；

使用所述空間特征計算模塊對每個所述第一目標圖像的每個像素特征進行計算，得到每個所述第一目標圖像的像素權重；

根據每個所述第一目標圖像以及與所述第一目標圖像對應的所述通道權重和所述像素權重確定增強后的目標圖像集。

3.根據權利要求1所述的方法，其特征在于，所述獲取樣本圖像的多個第一目標框，包括：

獲取樣本圖像的多個標注框，所述標注框包括標注框的類別信息；

根據所述多個標注框的類別信息確定所述多個第一目標框。

4.根據權利要求3所述的方法，其特征在于，所述將所述特征增強后的目標圖像集添加至所述樣本圖像中，得到增強后的圖像，包括：

獲取所述多個標注框在所述樣本圖像中的第一區域位置；

將所述特征增強后的目標圖像集添加至所述樣本圖像中除所述第一區域位置之外的目標區域位置，得到增強后的樣本圖像。

5.根據權利要求4所述的方法，其特征在于，所述方法還包括：

確定所述類別信息對應的標注框的數量少于閾值的至少一個標注框為第二目標框，所述第二目標框包括第二目標圖像；

所述將所述特征增強后的目標圖像集添加至所述樣本圖像中除所述第一區域位置之外的目標區域位置，得到增強后的圖像，包括：

將所述第二目標框包括的第二目標圖像和所述特征增強后的目標圖像集添加至所述樣本圖像中除所述第一區域位置之外的目標區域位置，得到增強后的圖像。

6.根據權利要求5所述的方法，其特征在于，所述將所述第二目標框包括的第二目標圖像和所述特征增強后的目標圖像集添加至所述樣本圖像中除所述第一區域位置之外的目標區域位置，得到增強后的圖像，包括：

將所述第二目標框包括的第二目標圖像和所述特征增強后的目標圖像集進行縮放、角度旋轉中的至少一種處理；

將經過處理的目標圖像集和第二目標圖像添加至所述樣本圖像中除所述第一區域位置之外的目標區域位置，得到增強后的圖像。

7.根據權利要求5所述的方法，其特征在于，所述第一目標框的數量占所述標注框的總數量的第一比值與所述標注框中除所述第一目標框和所述第二目標框之外的標注框的數量占所述總數量的第二比值滿足預設條件。

8.一種樣本圖像的處理裝置，其特征在于，所述裝置包括：

獲取模塊，用于獲取樣本圖像的多個第一目標框，其中，所述第一目標框包括第一目標圖像，所述第一目標圖像包括通道特征和空間特征；

處理模塊，用于使用預設的神經網絡中包括的通道特征計算模塊和空間特征計算模塊對每個所述第一目標框包括的第一目標圖像進行計算，得到特征增強后的目標圖像集；

所述處理模塊，還用于將所述特征增強后的目標圖像集添加至所述樣本圖像中，得到增強后的圖像。