[發明專利]目標檢測模型訓練方法、裝置及通信設備

在審

• 申請號: 202110264474.7
• 申請日: 2021-03-11
• 公開（公告）號: CN113034449A
• 公開（公告）日: 2021-06-25
• 發明（設計）人: 邵池;程駿;林燦然;郭渺辰;龐建新
• 申請（專利權）人: 深圳市優必選科技股份有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T5/00;G06T7/11
• 代理公司: 深圳中一聯合知識產權代理有限公司 44414
• 代理人: 甘瑩

1.一種目標檢測模型訓練方法，其特征在于，包括：

獲取第一圖像，所述第一圖像為經過人工標注的圖像，一張所述第一圖像唯一對應一個標注信息；

對所述第一圖像進行處理，得到與所述第一圖像不同的第二圖像，所述第二圖像的標注信息與所述第一圖像的標注信息相同；

根據所述第一圖像和所述第二圖像對待訓練的目標檢測模型進行訓練，得到訓練后的目標檢測模型。

2.如權利要求1所述的目標檢測模型訓練方法，其特征在于，所述對所述第一圖像進行處理，得到與所述第一圖像不同的第二圖像，包括：

對所述第一圖像的光線和/或色調進行處理，得到與所述第一圖像不同的第二圖像。

3.如權利要求2所述的目標檢測模型訓練方法，其特征在于，所述對所述第一圖像的光線和/或色調進行處理，得到與所述第一圖像不同的第二圖像，包括：

獲取第三圖像，所述第三圖像為純色的圖像；

根據所述第三圖像、預設的所述第三圖像的權重、所述第一圖像以及預設的所述第一圖像的權重確定所述第二圖像，所述第二圖像為光線和/或色調進行處理后的所述第一圖像。

4.如權利要求3所述的目標檢測模型訓練方法，其特征在于，所述根據所述第三圖像、預設的所述第三圖像的權重、所述第一圖像以及預設的所述第一圖像的權重確定所述第二圖像，包括：

根據所述第三圖像、預設的所述第三圖像的權重、所述第一圖像、預設的所述第一圖像的權重以及預設的亮度調整值確定所述第二圖像。

5.如權利要求1至4任一項所述的目標檢測模型訓練方法，其特征在于，所述標注信息包括人工標注的檢測框在所述第二圖像的位置信息，所述目標檢測模型訓練方法，還包括：

獲取第四圖像，所述第四圖像為沒有經過人工標注的圖像；

根據所述位置信息從所述第一圖像中裁剪出所述檢測框的區域，得到裁剪圖像；

將所述裁剪圖像與所述第四圖像進行融合，得到融合圖像，所述融合圖像的標注信息與所述第一圖像的標注信息相同；

所述根據所述第一圖像和所述第二圖像對待訓練的目標檢測模型進行訓練，得到訓練后的目標檢測模型，包括：

根據所述第一圖像、所述第二圖像以及所述融合圖像對待訓練的目標檢測模型進行訓練，得到訓練后的目標檢測模型。

6.如權利要求5所述的目標檢測模型訓練方法，其特征在于，所述將所述裁剪圖像與所述第四圖像進行融合，得到融合圖像，包括：

將所述第四圖像的尺寸調整為預設尺寸后，將所述裁剪圖像與尺寸調整后的所述第四圖像進行融合，得到融合圖像。

7.一種目標檢測方法，其特征在于，包括：

獲取待檢測的圖像；

采用如權利要求1至6任一項所述的訓練后的目標檢測模型對所述待檢測的圖像進行目標檢測，得到檢測結果。

8.一種目標檢測模型訓練裝置，其特征在于，包括：

第一圖像獲取單元，用于獲取第一圖像，所述第一圖像為經過人工標注的圖像，一張所述第一圖像唯一對應一個標注信息；

第二圖像獲取單元，用于對所述第一圖像進行處理，得到與所述第一圖像不同的第二圖像，所述第二圖像的標注信息與所述第一圖像的標注信息相同；

訓練單元，用于根據所述第一圖像和所述第二圖像對待訓練的目標檢測模型進行訓練，得到訓練后的目標檢測模型。

9.一種通信設備，包括存儲器、處理器以及存儲在所述存儲器中并可在所述處理器上運行的計算機程序，其特征在于，所述處理器執行所述計算機程序時實現如權利要求1至7任一項所述的方法。

10.一種計算機可讀存儲介質，所述計算機可讀存儲介質存儲有計算機程序，其特征在于，所述計算機程序被處理器執行時實現如權利要求1至7任一項所述的方法。