[發(fā)明專利]訓(xùn)練模型的方法、裝置、電子設(shè)備、存儲(chǔ)介質(zhì)及產(chǎn)品

在審

• 申請?zhí)枺?/td>: 202210109350.6
• 申請日: 2022-01-28
• 公開（公告）號: CN114445682A
• 公開（公告）日: 2022-05-06
• 發(fā)明（設(shè)計(jì)）人: 何斌
• 申請（專利權(quán)）人: 北京百度網(wǎng)訊科技有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/44;G06V40/16;G06V20/00;G06V10/26;G06V10/82;G06K9/62;G06N3/08
• 代理公司: 北京鉦霖知識(shí)產(chǎn)權(quán)代理有限公司 11722
• 代理人: 李英艷;楊繼成

1.一種訓(xùn)練模型的方法，包括：

獲取第一圖像集，并調(diào)用第一模型以及第二模型，通過所述第一模型和所述第二模型分別對所述第一圖像集進(jìn)行目標(biāo)檢測，所述第一模型的目標(biāo)檢測準(zhǔn)確度小于所述第二模型的目標(biāo)檢測準(zhǔn)確度；

確定所述第一模型對所述第一圖像集進(jìn)行目標(biāo)檢測的第一目標(biāo)檢測結(jié)果，并確定所述第二模型對所述第一圖像集進(jìn)行目標(biāo)檢測的第二目標(biāo)檢測結(jié)果；

基于所述第一目標(biāo)檢測結(jié)果和所述第二目標(biāo)檢測結(jié)果，確定誤檢圖像；

生成包括所述誤檢圖像的第二圖像集，并基于所述第二圖像集訓(xùn)練所述第一模型。

2.根據(jù)權(quán)利要求1所述的方法，其中，所述生成包括所述誤檢圖像的第二圖像集，包括：

在目標(biāo)數(shù)據(jù)庫中，獲取多張目標(biāo)圖像，所述目標(biāo)圖像中包括有所述第一模型和所述第二模型進(jìn)行目標(biāo)檢測的檢測目標(biāo)；

將所述多張目標(biāo)圖像與所述誤檢圖像進(jìn)行組合，生成包括組合后圖像的第二圖像集。

3.根據(jù)權(quán)利要求2所述的方法，其中，所述將所述多張目標(biāo)圖像與所述誤檢圖像進(jìn)行組合，包括：

對所述誤檢圖像進(jìn)行預(yù)處理，得到誤檢區(qū)域圖像；

對所述多張目標(biāo)圖像進(jìn)行預(yù)處理，得到多張目標(biāo)區(qū)域圖像；

組合所述誤檢區(qū)域圖像以及所述多張目標(biāo)區(qū)域圖像，得到組合后圖像，其中，所述組合后圖像的中心為所述誤檢區(qū)域圖像，所述多張目標(biāo)區(qū)域圖像位于所述誤檢區(qū)域圖像周邊。

4.根據(jù)權(quán)利要求3所述的方法，其中，所述誤檢區(qū)域圖像與所述多張目標(biāo)區(qū)域圖像以特定方式呈現(xiàn)，所述特定方式包括九宮格形式、十字形式和X形式。

5.根據(jù)權(quán)利要求3所述的方法，其中，所述對所述誤檢圖像進(jìn)行預(yù)處理，得到誤檢區(qū)域圖像，包括：

擴(kuò)展所述誤檢圖像的尺寸，并在所述誤檢圖像中確定誤檢區(qū)域；

在擴(kuò)展尺寸后的誤檢圖像中裁剪得到包括誤檢區(qū)域的誤檢區(qū)域圖像。

6.根據(jù)權(quán)利要求1-5中任意一項(xiàng)所述的方法，其中，所述基于所述第一目標(biāo)檢測結(jié)果和所述第二目標(biāo)檢測結(jié)果，確定誤檢圖像，包括：

確定第一模型對第一圖像集中第一圖像的第一目標(biāo)檢測結(jié)果，并確定所述第二模型對所述第一圖像的第二目標(biāo)檢測結(jié)果；

確定所述第一目標(biāo)檢測結(jié)果和第二目標(biāo)檢測結(jié)果的交并比；

若所述交并比小于閾值，則將所述第一圖像確定為誤檢圖像。

7.一種訓(xùn)練模型的裝置，包括：

目標(biāo)檢測單元，用于獲取第一圖像集，并調(diào)用第一模型以及第二模型，通過所述第一模型和所述第二模型分別對所述第一圖像集進(jìn)行目標(biāo)檢測，所述第一模型的目標(biāo)檢測準(zhǔn)確度小于所述第二模型的目標(biāo)檢測準(zhǔn)確度；

確定單元，用于確定所述第一模型對所述第一圖像集進(jìn)行目標(biāo)檢測的第一目標(biāo)檢測結(jié)果，并確定所述第二模型對所述第一圖像集進(jìn)行目標(biāo)檢測的第二目標(biāo)檢測結(jié)果，還用于基于所述第一目標(biāo)檢測結(jié)果和所述第二目標(biāo)檢測結(jié)果，確定誤檢圖像；

生成單元，用于生成包括所述誤檢圖像的第二圖像集；

訓(xùn)練單元，用于基于所述第二圖像集訓(xùn)練所述第一模型。

8.根據(jù)權(quán)利要求7所述的裝置，其中，所述生成單元，還用于：

在目標(biāo)數(shù)據(jù)庫中，獲取多張目標(biāo)圖像，所述目標(biāo)圖像中包括有所述第一模型和所述第二模型進(jìn)行目標(biāo)檢測的檢測目標(biāo)；

將所述多張目標(biāo)圖像與所述誤檢圖像進(jìn)行組合，生成包括組合后圖像的第二圖像集。

9.根據(jù)權(quán)利要求8所述的裝置，其中，所述生成單元，還用于：

對所述誤檢圖像進(jìn)行預(yù)處理，得到誤檢區(qū)域圖像；

對所述多張目標(biāo)圖像進(jìn)行預(yù)處理，得到多張目標(biāo)區(qū)域圖像；

組合所述誤檢區(qū)域圖像以及所述多張目標(biāo)區(qū)域圖像，得到組合后圖像，其中，所述組合后圖像的中心為所述誤檢區(qū)域圖像，所述多張目標(biāo)區(qū)域圖像位于所述誤檢區(qū)域圖像周邊。