[發(fā)明專利]圖像識別模型的遷移方法、裝置、設(shè)備及存儲介質(zhì)

有效

• 申請?zhí)枺?/td>: 202110398557.5
• 申請日: 2021-04-14
• 公開（公告）號: CN112801236B
• 公開（公告）日: 2022-03-11
• 發(fā)明（設(shè)計）人: 劉暢
• 申請（專利權(quán)）人: 騰訊科技（深圳）有限公司
• 主分類號: G06V10/80
• 分類號: G06V10/80;G06V10/774;G06K9/62;G06N3/08
• 代理公司: 北京三高永信知識產(chǎn)權(quán)代理有限責(zé)任公司 11138
• 代理人: 祝亞男

本申請公開了一種圖像識別模型的遷移方法、裝置、設(shè)備及存儲介質(zhì)，涉及人工智能技術(shù)領(lǐng)域。所述方法包括：獲取源域圖像樣本和目標(biāo)域圖像樣本的頭部特征圖；基于兩個樣本的頭部特征圖確定兩個樣本的對象預(yù)測標(biāo)注；基于兩個樣本的特征圖之間的差異信息以及源域圖像樣本的對象預(yù)測標(biāo)注與其真值標(biāo)注之間的差異信息，確定模型訓(xùn)練損失；根據(jù)模型訓(xùn)練損失調(diào)整圖像識別模型的參數(shù)，得到遷移后圖像識別模型。本申請通過基于頭部特征圖實現(xiàn)了單階段的領(lǐng)域遷移，能夠提高模型的遷移效率和模型的泛化性。另外，本申請可利用帶真值標(biāo)注的仿真數(shù)據(jù)和無真值標(biāo)注的真實場景數(shù)據(jù)，將模型遷移至真實場景中運用。例如，本申請可用于自動駕駛場景的模型領(lǐng)域遷移任務(wù)。

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

本申請實施例涉及人工智能技術(shù)領(lǐng)域，特別涉及一種圖像識別模型的遷移方法、裝置、設(shè)備及存儲介質(zhì)。

背景技術(shù)

隨著人工智能技術(shù)的發(fā)展，其在領(lǐng)域遷移的研究和應(yīng)用也越來越多。領(lǐng)域遷移用于將模型從源領(lǐng)域的數(shù)據(jù)分布遷移到目標(biāo)領(lǐng)域的數(shù)據(jù)分布，遷移后模型可以用于執(zhí)行目標(biāo)領(lǐng)域?qū)?yīng)的任務(wù)。

以圖像識別模型為例，相關(guān)技術(shù)通過基于雙階段的圖像識別方法來實現(xiàn)領(lǐng)域遷移，即針對目標(biāo)領(lǐng)域設(shè)計一組合理的錨點集（即候選框集），基于錨點集對圖像識別模型的參數(shù)進(jìn)行細(xì)調(diào)，以得到適用于目標(biāo)領(lǐng)域的遷移后圖像識別模型。

然而，相關(guān)技術(shù)在無法獲取目標(biāo)域圖像樣本的真實標(biāo)注的情況下，是無法針對目標(biāo)領(lǐng)域設(shè)計出合理的錨點集的，圖像識別模型的泛化性不強。

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

本申請實施例提供了一種圖像識別模型的遷移方法、裝置、設(shè)備及存儲介質(zhì)，能夠提高模型的遷移效率以及模型的泛化性。所述技術(shù)方案包括如下內(nèi)容。

根據(jù)本申請實施例的一個方面，提供了一種圖像識別模型的遷移方法，所述方法包括：

獲取源域圖像樣本和目標(biāo)域圖像樣本，所述源域圖像樣本是指源領(lǐng)域?qū)?yīng)的帶真值標(biāo)注的圖像樣本，所述目標(biāo)域圖像樣本是指待遷移至的目標(biāo)領(lǐng)域?qū)?yīng)的不帶真值標(biāo)注的圖像樣本；

通過待遷移的圖像識別模型對所述源域圖像樣本和所述目標(biāo)域圖像樣本分別進(jìn)行特征提取和融合處理，得到所述源域圖像樣本的頭部特征圖和所述目標(biāo)域圖像樣本的頭部特征圖；其中，所述頭部特征圖用于指示圖像樣本中的目標(biāo)對象的類別信息和位置信息；

基于所述源域圖像樣本的頭部特征圖確定所述源域圖像的對象預(yù)測標(biāo)注，以及基于所述目標(biāo)域圖像樣本的頭部特征圖確定所述目標(biāo)域圖像樣本的對象預(yù)測標(biāo)注，所述對象預(yù)測標(biāo)注用于預(yù)測標(biāo)注所述目標(biāo)對象；

基于所述源域圖像樣本和所述目標(biāo)域圖像樣本在圖像識別過程中得到的特征圖之間的差異信息，以及所述源域圖像樣本的對象預(yù)測標(biāo)注與其真值標(biāo)注之間的差異信息，確定模型訓(xùn)練損失；其中，所述特征圖包括所述頭部特征圖；

根據(jù)所述模型訓(xùn)練損失調(diào)整所述圖像識別模型的參數(shù)，得到適用于所述目標(biāo)領(lǐng)域的遷移后圖像識別模型。

根據(jù)本申請實施例的一個方面，提供了一種圖像識別模型的遷移裝置，所述裝置包括：

圖像樣本獲取模塊，用于獲取源域圖像樣本和目標(biāo)域圖像樣本，所述源域圖像樣本是指源領(lǐng)域?qū)?yīng)的帶真值標(biāo)注的圖像樣本，所述目標(biāo)域圖像樣本是指待遷移至的目標(biāo)領(lǐng)域?qū)?yīng)的不帶真值標(biāo)注的圖像樣本；

頭部特征圖獲取模塊，用于通過待遷移的圖像識別模型對所述源域圖像樣本和所述目標(biāo)域圖像樣本分別進(jìn)行特征提取和融合處理，得到所述源域圖像樣本的頭部特征圖和所述目標(biāo)域圖像樣本的頭部特征圖；其中，所述頭部特征圖用于指示圖像樣本中的目標(biāo)對象的類別信息和位置信息；

預(yù)測標(biāo)注獲取模塊，用于基于所述源域圖像樣本的頭部特征圖確定所述源域圖像的對象預(yù)測標(biāo)注，以及基于所述目標(biāo)域圖像樣本的頭部特征圖確定所述目標(biāo)域圖像樣本的對象預(yù)測標(biāo)注，所述對象預(yù)測標(biāo)注用于預(yù)測標(biāo)注所述目標(biāo)對象；