一種圖像識別模型訓(xùn)練方法、裝置、設(shè)備及介質(zhì)

基本信息

申請?zhí)?/td> CN202111666239.9 申請日 -
公開(公告)號 CN114359547A 公開(公告)日 2022-04-15
申請公布號 CN114359547A 申請公布日 2022-04-15
分類號 G06V10/25(2022.01)I;G06V10/774(2022.01)I;G06V10/74(2022.01)I;G06V10/82(2022.01)I;G06N3/04(2006.01)I 分類 計算;推算;計數(shù);
發(fā)明人 劉冰;百曉;陳科;薛姬榮;李軍;周雷;王晨;張亮;江金壽;田建輝;王曉悅;葉金華;安辰 申請(專利權(quán))人 中國兵器科學(xué)研究院
代理機(jī)構(gòu) 北京同達(dá)信恒知識產(chǎn)權(quán)代理有限公司 代理人 王麗莉
地址 100089北京市海淀區(qū)車道溝十號院中國兵器科學(xué)研究院
法律狀態(tài) -

摘要

摘要 本申請實施例提供了一種圖像識別模型訓(xùn)練方法、裝置、設(shè)備及介質(zhì),由于本申請中,原始圖像識別模型輸出了每個樣本圖像對中參考圖像樣本對應(yīng)的第一預(yù)估圖像及第二預(yù)估圖像,并分別根據(jù)第一預(yù)估圖像中對應(yīng)像素點的像素值確定了第一子損失值,根據(jù)第二預(yù)估圖像中對應(yīng)像素點的像素值,確定第二損失值,進(jìn)行了像素級對比學(xué)習(xí),還輸出了每個樣本圖像對中參考圖像樣本對應(yīng)的第一特征向量及每個樣本圖像對中目標(biāo)圖像樣本對應(yīng)的第二特征向量,并分別根據(jù)第一特征向量以及第二特征向量,確定第三子損失值,進(jìn)行了圖像級對比學(xué)習(xí),由于進(jìn)行了像素級對比和圖像級對比學(xué)習(xí),提高了圖像識別模型的精度,及圖像識別模型提取出的圖像中物體特征的顯著性或區(qū)分性。