[發明專利]圖像數據處理方法及裝置、目標檢測模型訓練方法及裝置

在審

• 申請號: 202011507842.8
• 申請日: 2020-12-18
• 公開（公告）號: CN112633355A
• 公開（公告）日: 2021-04-09
• 發明（設計）人: 陳亮宇
• 申請（專利權）人: 北京邁格威科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 北京鉦霖知識產權代理有限公司 11722
• 代理人: 李英艷;李志新

1.一種圖像數據處理方法，其中，所述圖像數據處理方法包括：

獲取第一圖像集，所述第一圖像集包括多個第一圖像及其對應的第一標注信息，其中，所述第一圖像包括至少一個第一目標，所述第一標注信息包括所述第一目標對應的第一邊界框和類別；

基于所述第一圖像集，訓練數據處理模型；

獲取第二圖像集，所述第二圖像集包括多個第二圖像及其對應的第二標注信息，其中，所述第二圖像包括至少一個第二目標，所述第二標注信息包括所述第二目標對應的第二標注點和類別；

基于所述第二圖像集，通過完成訓練的所述數據處理模型，得到第三圖像集，所述第三圖像集包括所述多個第二圖像及其對應的第三標注信息，其中，所述第三標注信息包括所述第二目標對應的第二邊界框和類別。

2.根據權利要求1所述的圖像數據處理方法，其中，所述基于所述第一圖像集，訓練數據處理模型，包括：

在所述第一圖像的每個第一邊界框內生成對應所述第一目標的第一標注點；

基于所述第一圖像、以及所述第一圖像中所述第一目標對應的第一標注點及類別，通過待訓練的數據處理模型得到對應于所述第一目標的第一預測框；

基于對應于同一第一目標的所述第一預測框與所述第一邊界框得到損失值；

基于所述損失值調整待訓練的數據處理模型的參數，得到完成訓練的所述數據處理模型。

3.根據權利要求2所述的圖像數據處理方法，其中，所述基于所述第一圖像集，訓練數據處理模型，還包括：

對所述第一標注點的坐標進行歸一化。

4.根據權利要求2或3所述的圖像數據處理方法，其中，所述損失值包括以下一種或多種，L1損失、IOU損失及GIOU損失。

5.根據權利要求2或3所述的圖像數據處理方法，其中，所述基于所述第二圖像集，通過完成訓練的所述數據處理模型，得到第三圖像集，包括：

基于所述第二目標對應的第二標注點和類別，通過完成訓練的所述數據處理模型，得到所述第二目標對應的第二邊界框。

6.一種目標檢測模型訓練方法，其中，所述目標檢測模型用于圖像目標檢測，所述目標檢測模型訓練方法包括：

獲取訓練集，所述訓練集至少包括采用如權利要求1-5所述的圖像數據處理方法得到的所述第三圖像集；

基于所述訓練集，訓練待訓練的目標檢測模型，得到訓練完成的所述目標檢測模型。

7.根據權利要求6所述的目標檢測模型訓練方法，其中，所述訓練集還包括所述第一圖像集。

8.根據權利要求7所述的目標檢測模型訓練方法，其中，所述基于所述訓練集，訓練待訓練的目標檢測模型，得到訓練完成的所述目標檢測模型，包括：

基于所述第一圖像或第二圖像，通過待訓練的目標檢測模型，得到所述第一目標或第二目標的預測框；

基于對應于同一第一圖像的預測框和第一邊界框，或對應于同一第二圖像的預測框和第二邊界框，得到邊界框損失值；

基于所述邊界框損失值調整待訓練的目標檢測模型的參數，得到完成訓練的目標檢測模型。

9.一種目標檢測方法，其中，所述目標檢測方法包括：

獲取圖像，所述圖像包括至少一個待檢測目標；

通過目標檢測模型，確定所述待檢測目標的邊界框，其中，所述目標檢測模型通過如權利要求6-8任一項所述的目標檢測模型訓練方法訓練得到。