[發明專利]訓練目標檢測模型和構建樣本集的方法、裝置及存儲介質

有效

• 申請號: 202210217860.5
• 申請日: 2022-03-08
• 公開（公告）號: CN114359676B
• 公開（公告）日: 2022-07-19
• 發明（設計）人: 陳文晶;王堅;李兵;余昊楠;胡衛明
• 申請（專利權）人: 人民中科（濟南）智能技術有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/764;G06K9/62;G06T7/70
• 代理公司: 北京萬思博知識產權代理有限公司 11694
• 代理人: 劉冀

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

利用第一訓練樣本集對第一目標檢測模型進行訓練，其中所述訓練樣本集

包括第一圖像集合和第一標注集合，所述第一圖像集合包括多個第一圖像，所

述第一標注集合包括分別與所述多個第一圖像對應的第一標注信息，并且每個

第一標注信息包括第一邊界框位置信息以及與所述第一邊界框位置信息對應的

類別信息；

利用所述第一目標檢測模型確定與第二圖像集合對應的第二標注集合，所

述第二圖像集合包括多個第二圖像，所述第二標注集合包括分別與所述多個第

二圖像對應的第二標注信息，并且每個第二標注信息包括第二邊界框位置信息

以及與所述第二邊界框位置信息對應的類別信息；以及

將所述第一圖像與相應的第一標注信息進行配對，將所述第二圖像與相應

的第二標注信息進行配對，并且利用配對后的圖像和標注信息對待訓練的目標

檢測模型進行訓練，并且其中

將所述第一圖像與相應的第一標注信息進行配對，將所述第二圖像與相應

的第二標注信息進行配對，并且利用配對后的圖像和標注信息，對待訓練的目

標檢測模型進行訓練的操作，包括：

對所述第一圖像集合和所述第二圖像集合中的圖像進行組合，得到對待訓

練的目標檢測模型進行訓練的圖像批次；以及

將所述圖像批次中的第一圖像與相應的第一標注信息配對，將所述圖像批

次中的第二圖像與相應的第二標注信息配對，并且利用配對后的圖像和標注信

息，對待訓練的目標檢測模型進行訓練，并且

利用所述第一目標檢測模型確定與第二圖像集合對應的第二標注集合的操

作，包括：

利用設置于所述第一目標檢測模型的錨框，通過所述第一目標檢測模型對

所述多個第二圖像進行目標檢測，確定所述第二標注集合的第一標注子集，其

中所述第一標注子集包括多個第三標注信息，所述第三標注信息分別與所述多

個第二圖像對應，并且每個第三標注信息包括根據所述錨框確定的第三邊界框

位置信息以及與所述第三邊界框位置信息對應的類別信息；

對所述多個第二圖像進行隨機圖像塊采樣處理，確定與所述多個第二圖像

對應的圖像塊的圖像塊邊界框信息；以及

利用設置于所述第一目標檢測模型的錨框以及所述圖像塊邊界框信息，通

過所述第一目標檢測模型對所述多個第二圖像進行目標檢測，確定所述第二標

注集合的第二標注子集，其中所述第二標注子集包括多個第四標注信息，所述

第四標注信息分別與所述多個第二圖像對應，并且每個第四標注信息包括根據

所述錨框以及所述圖像塊邊界框信息確定的第三邊界框位置信息以及與所述第

三邊界框位置信息對應的類別信息。

2.根據權利要求1 所述的方法，其特征在于，對所述第一圖像集合和所述

第二圖像集合中的圖像進行組合，得到對待訓練的目標檢測模型進行訓練的圖

像批次的操作，包括：

對所述第一圖像集合和所述第二圖像集合中的圖像進行圖像增強處理；以

及

將圖像增強處理后的圖像進行組合，得到所述圖像批次。