[發(fā)明專利]用于對象識(shí)別模型的訓(xùn)練方法、裝置及計(jì)算機(jī)存儲(chǔ)介質(zhì)

在審

• 申請?zhí)枺?/td>: 202110272497.2
• 申請日: 2021-03-12
• 公開（公告）號(hào): CN115081500A
• 公開（公告）日: 2022-09-20
• 發(fā)明（設(shè)計(jì)）人: 黃詩盛
• 申請（專利權(quán)）人: 深圳海翼智新科技有限公司
• 主分類號(hào): G06K9/62
• 分類號(hào): G06K9/62;G06N3/04;G06N3/08
• 代理公司: 深圳市威世博知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 44280
• 代理人: 黎堅(jiān)怡

本發(fā)明公開一種用于對象識(shí)別模型的訓(xùn)練方法、裝置以及計(jì)算機(jī)存儲(chǔ)介質(zhì)，該方法包括：獲取對象數(shù)據(jù)訓(xùn)練集和對象數(shù)據(jù)測試集，對象數(shù)據(jù)訓(xùn)練集包括訓(xùn)練背景和標(biāo)注的訓(xùn)練對象框，對象數(shù)據(jù)測試集包括測試背景和標(biāo)注的測試對象框；利用對象數(shù)據(jù)訓(xùn)練集訓(xùn)練對象識(shí)別模型；利用對象數(shù)據(jù)測試集測試對象識(shí)別模型并且獲得對應(yīng)的召回率、漏檢的測試對象框和誤檢的測試背景框；以及響應(yīng)于召回率小于召回率閾值，將漏檢的測試對象框加入對象數(shù)據(jù)訓(xùn)練集作為標(biāo)注的訓(xùn)練對象框，將誤檢的測試背景框加入對象數(shù)據(jù)訓(xùn)練集作為訓(xùn)練背景，返回利用對象數(shù)據(jù)訓(xùn)練集訓(xùn)練對象識(shí)別模型的步驟以進(jìn)行迭代訓(xùn)練。本發(fā)明可以在不增加對象識(shí)別模型復(fù)雜度的情況下，有效提升模型精度。

技術(shù)領(lǐng)域

本發(fā)明涉及對象識(shí)別領(lǐng)域，特別是涉及一種用于對象識(shí)別模型的訓(xùn)練方法、裝置及計(jì)算機(jī)存儲(chǔ)介質(zhì)。

背景技術(shù)

以人臉識(shí)別為代表的對象識(shí)別是計(jì)算機(jī)視覺領(lǐng)域中應(yīng)用最廣泛的一種技術(shù)，并且是諸如自動(dòng)駕駛、智能支付、安全監(jiān)測、智能家居等領(lǐng)域的基礎(chǔ)技術(shù)。如何提高對象識(shí)別模型的精度是工業(yè)界一直追求的目標(biāo)。

工業(yè)界往往通過增加模型表達(dá)能力、優(yōu)化損失函數(shù)、增加關(guān)鍵點(diǎn)作為監(jiān)督信息等方法優(yōu)化對象識(shí)別的模型。但是這些方法在提高模型精度的同時(shí)也急劇增加了模型的復(fù)雜度。

發(fā)明內(nèi)容

本發(fā)明提供一種用于對象識(shí)別模型的訓(xùn)練方法、裝置及計(jì)算機(jī)存儲(chǔ)介質(zhì)，以解決現(xiàn)有技術(shù)中提高模型精度的同時(shí)，模型復(fù)雜度急劇增加的問題。

為解決上述技術(shù)問題，本發(fā)明采用的一個(gè)技術(shù)方案是一種用于對象識(shí)別模型的訓(xùn)練方法。該方法包括：獲取對象數(shù)據(jù)訓(xùn)練集和對象數(shù)據(jù)測試集，對象數(shù)據(jù)訓(xùn)練集包括訓(xùn)練背景和標(biāo)注的訓(xùn)練對象框，對象數(shù)據(jù)測試集包括測試背景和標(biāo)注的測試對象框；利用對象數(shù)據(jù)訓(xùn)練集訓(xùn)練對象識(shí)別模型；利用對象數(shù)據(jù)測試集測試對象識(shí)別模型并且獲得對應(yīng)的召回率、漏檢的測試對象框和誤檢的測試背景框；以及響應(yīng)于召回率小于召回率閾值，將漏檢的測試對象框加入對象數(shù)據(jù)訓(xùn)練集作為標(biāo)注的訓(xùn)練對象框，將誤檢的測試背景框加入對象數(shù)據(jù)訓(xùn)練集作為訓(xùn)練背景，返回利用對象數(shù)據(jù)訓(xùn)練集訓(xùn)練對象識(shí)別模型的步驟以進(jìn)行迭代訓(xùn)練。

為解決上述技術(shù)問題，本發(fā)明采用的另一個(gè)技術(shù)方案是一種用于對象識(shí)別模型的訓(xùn)練裝置。該裝置包括獲取模塊、訓(xùn)練模塊和測試模塊。獲取模塊配置成獲取對象數(shù)據(jù)訓(xùn)練集和對象數(shù)據(jù)測試集，對象數(shù)據(jù)訓(xùn)練集包括標(biāo)注的訓(xùn)練對象框和訓(xùn)練背景，對象數(shù)據(jù)測試集包括標(biāo)注的測試對象框和測試背景。訓(xùn)練模塊配置成利用對象數(shù)據(jù)訓(xùn)練集訓(xùn)練對象識(shí)別模型。測試模塊配置成利用對象數(shù)據(jù)測試集測試對象識(shí)別模型并且獲得對應(yīng)的召回率、漏檢的測試對象框和誤檢的測試背景框。其中，響應(yīng)于召回率小于召回率閾值，將漏檢的測試對象框加入對象數(shù)據(jù)訓(xùn)練集作為訓(xùn)練對象框，將誤檢的測試背景框加入對象數(shù)據(jù)訓(xùn)練集作為訓(xùn)練背景，返回利用對象數(shù)據(jù)訓(xùn)練集訓(xùn)練對象識(shí)別模型的步驟以進(jìn)行迭代訓(xùn)練。

為解決上述技術(shù)問題，本發(fā)明采用的另一個(gè)技術(shù)方案是一種用于對象識(shí)別模型的訓(xùn)練裝置。該裝置包括處理器和存儲(chǔ)器。存儲(chǔ)器中存儲(chǔ)有計(jì)算機(jī)程序。處理器用于執(zhí)行計(jì)算機(jī)程序以實(shí)現(xiàn)上述用于對象識(shí)別模型的訓(xùn)練方法。

為解決上述技術(shù)問題，本發(fā)明采用的又一個(gè)技術(shù)方案是一種非暫態(tài)計(jì)算機(jī)存儲(chǔ)介質(zhì)。該非暫態(tài)計(jì)算機(jī)存儲(chǔ)介質(zhì)存儲(chǔ)有計(jì)算機(jī)程序。該計(jì)算機(jī)程序被執(zhí)行時(shí)實(shí)現(xiàn)上述用于對象識(shí)別模型的訓(xùn)練方法。

區(qū)別于現(xiàn)有技術(shù)，本發(fā)明通過上述方法、裝置和存儲(chǔ)介質(zhì)，實(shí)現(xiàn)了在不增加對象識(shí)別模型復(fù)雜度的情況下，有效提升模型精度的技術(shù)效果。

附圖說明

為更清楚地說明本發(fā)明實(shí)施例中的技術(shù)方案，下面將對實(shí)施例描述中所需要使用的附圖作簡單地介紹，顯而易見地，下面描述中的附圖僅僅是本發(fā)明的一些實(shí)施例，對本領(lǐng)域普通技術(shù)人員來講，在不付出創(chuàng)造性勞動(dòng)的前提下，還可以根據(jù)這些附圖獲得其他的附圖。

圖1是本發(fā)明用于對象識(shí)別模型的訓(xùn)練方法的流程示意圖；

圖2示出對象數(shù)據(jù)測試集中包含漏檢的測試對象框的一個(gè)示例圖像；