一種目標(biāo)檢測模型訓(xùn)練方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)
基本信息
| 申請(qǐng)?zhí)?/td> | CN202210312655.7 | 申請(qǐng)日 | - |
| 公開（公告）號(hào) | CN114663728A | 公開（公告）日 | 2022-06-24 |
| 申請(qǐng)公布號(hào) | CN114663728A | 申請(qǐng)公布日 | 2022-06-24 |
| 分類號(hào) | G06V10/774（2022.01）I;G06K9/62（2022.01）I;G06N3/04（2006.01）I;G06T7/11（2017.01）I;G06T7/62（2017.01）I | 分類 | 計(jì)算；推算；計(jì)數(shù); |
| 發(fā)明人 | 王堃;顧杏春;林業(yè)偉;王軍 | 申請(qǐng)（專利權(quán)）人 | 南京大牛信息科技有限公司 |
| 代理機(jī)構(gòu) | 北京超凡宏宇專利代理事務(wù)所（特殊普通合伙） | 代理人 | - |
| 地址 | 210000江蘇省南京市玄武區(qū)天山路39號(hào) | ||
| 法律狀態(tài) | - | ||
摘要
| 摘要 | 本發(fā)明的實(shí)施例提供了一種目標(biāo)檢測模型訓(xùn)練方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)，方法包括：將待調(diào)整圖像輸入至待訓(xùn)練的目標(biāo)檢測模型中，輸出目標(biāo)預(yù)測框；獲取目標(biāo)預(yù)測框?qū)?yīng)的標(biāo)準(zhǔn)框；確定目標(biāo)預(yù)測框與標(biāo)準(zhǔn)框的交疊率；基于交疊率，計(jì)算目標(biāo)預(yù)測框與標(biāo)準(zhǔn)框的目標(biāo)交疊率；基于目標(biāo)交疊率，計(jì)算目標(biāo)預(yù)測框與標(biāo)準(zhǔn)框之間的損失；將損失輸入目標(biāo)檢測模型，以獲得更新后的目標(biāo)檢測模型；返回執(zhí)行確定目標(biāo)預(yù)測框與標(biāo)準(zhǔn)框的交疊率至將損失輸入目標(biāo)檢測模型，以獲得更新后的目標(biāo)檢測模型的步驟，直至損失收斂為止。通過基于目標(biāo)預(yù)測框和標(biāo)準(zhǔn)框的交疊率，計(jì)算目標(biāo)交疊率，改變現(xiàn)有對(duì)目標(biāo)檢測模型的損失函數(shù)，從而提升目標(biāo)檢測模型識(shí)別的準(zhǔn)確率。 |
