[發(fā)明專利]模型訓(xùn)練及圖像識別方法和裝置、設(shè)備及存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202210107742.9
• 申請日: 2022-01-28
• 公開（公告）號: CN114445681A
• 公開（公告）日: 2022-05-06
• 發(fā)明（設(shè)計）人: 唐詩翔;朱烽;趙瑞
• 申請（專利權(quán)）人: 上海商湯智能科技有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/776;G06V10/74;G06K9/62;G06V10/40
• 代理公司: 北京派特恩知識產(chǎn)權(quán)代理有限公司 11270
• 代理人: 馬麗;胡春光

1.一種模型訓(xùn)練方法，其特征在于，所述方法包括：

獲取包含第一對象的第一圖像樣本；

利用待訓(xùn)練的第一模型的第一網(wǎng)絡(luò)，對所述第一圖像樣本進行特征提取，得到所述第一對象的第一特征；

利用所述第一模型的第二網(wǎng)絡(luò)，基于至少一個第二對象的第二特征，對所述第一特征進行更新，得到所述第一特征對應(yīng)的第一目標(biāo)特征，每一所述第二對象與所述第一對象的相似度不小于第一閾值；

基于所述第一目標(biāo)特征，確定目標(biāo)損失值；

基于所述目標(biāo)損失值，對所述第一模型的模型參數(shù)進行至少一次更新，得到訓(xùn)練后的所述第一模型。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述第一圖像樣本包括標(biāo)簽信息，所述第一模型中包括第一特征記憶庫，所述第一特征記憶庫中包括屬于至少一個對象的至少一個特征；

所述基于所述第一目標(biāo)特征，確定目標(biāo)損失值，包括：

基于所述第一目標(biāo)特征和所述標(biāo)簽信息，確定第一損失值；

基于所述第一目標(biāo)特征和所述第一特征記憶庫中的至少一個對象的至少一個特征，確定第二損失值；

基于所述第一損失值和所述第二損失值，確定所述目標(biāo)損失值。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述基于所述第一目標(biāo)特征和所述第一特征記憶庫中的至少一個對象的至少一個特征，確定第二損失值，包括：

從所述第一特征記憶庫中的至少一個對象的至少一個特征中，確定所述第一對象的第一特征中心和至少一個所述第二對象的第二特征中心；

基于所述第一目標(biāo)特征、所述第一特征中心和每一所述第二特征中心，確定所述第二損失值。

4.根據(jù)權(quán)利要求2或3所述的方法，其特征在于，所述第一特征記憶庫中包括屬于至少一個對象的特征集，每一所述特征集包括所屬對象的至少一個特征；所述方法還包括：

基于所述第一目標(biāo)特征，更新所述第一特征記憶庫中屬于所述第一對象的特征集。

5.根據(jù)權(quán)利要求1至4中任一項所述的方法，其特征在于，

所述獲取包含第一對象的第一圖像樣本，包括：獲取包含第一對象的第一子圖像和第二子圖像，所述第二子圖像是對第一子圖像至少進行遮擋處理后的圖像；

所述利用待訓(xùn)練的第一模型的第一網(wǎng)絡(luò)，對所述第一圖像樣本進行特征提取，得到所述第一對象的第一特征，包括：利用待訓(xùn)練的第一模型的第一網(wǎng)絡(luò)，對所述第一子圖像進行特征提取，得到所述第一對象的第一子特征，并對所述第二子圖像進行特征提取，得到所述第一對象的第二子特征；

所述利用所述第一模型的第二網(wǎng)絡(luò)，基于至少一個第二對象的第二特征，對所述第一特征進行更新，得到所述第一特征對應(yīng)的第一目標(biāo)特征，包括：利用所述第一模型的第二網(wǎng)絡(luò)，基于至少一個第二對象的第二特征，對所述第一子特征和所述第二子特征分別進行更新，得到所述第一子特征對應(yīng)的第一目標(biāo)子特征和所述第二子特征對應(yīng)的第二目標(biāo)子特征；

所述基于所述第一目標(biāo)特征，確定目標(biāo)損失值，包括：基于所述第一目標(biāo)子特征和所述第二目標(biāo)子特征，確定所述目標(biāo)損失值。

6.根據(jù)權(quán)利要求5所述的方法，其特征在于，所述基于所述第一目標(biāo)子特征和所述第二目標(biāo)子特征，確定所述目標(biāo)損失值，包括：

基于所述第一目標(biāo)子特征和所述第二目標(biāo)子特征，確定第一目標(biāo)損失值；

基于所述第一子特征和所述第二子特征，確定第二目標(biāo)損失值；

基于所述第一目標(biāo)損失值和所述第二目標(biāo)損失值，確定目標(biāo)損失值。

7.根據(jù)權(quán)利要求6所述的方法，其特征在于，所述獲取包含第一對象的第一子圖像和第二子圖像，包括：

獲取包含第一對象的第一子圖像；

基于預(yù)設(shè)的遮擋集，對所述第一子圖像至少進行遮擋處理，得到所述第二子圖像，所述遮擋集包括至少一個遮擋圖像。