基于圖像數(shù)據(jù)增強(qiáng)的圖像訓(xùn)練集生成方法和模型訓(xùn)練方法

基本信息

申請(qǐng)?zhí)?/td> CN202110617897.2 申請(qǐng)日 -
公開(kāi)（公告）號(hào) CN113361588A 公開(kāi)（公告）日 2021-09-07
申請(qǐng)公布號(hào) CN113361588A 申請(qǐng)公布日 2021-09-07
分類(lèi)號(hào) G06K9/62（2006.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類(lèi) 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 陳映;曹松;任必為 申請(qǐng)（專(zhuān)利權(quán)）人 北京文安智能技術(shù)股份有限公司
代理機(jī)構(gòu) - 代理人 -
地址 100094北京市海淀區(qū)豐豪東路9號(hào)院2號(hào)樓5單元801
法律狀態(tài) -

摘要

摘要 本發(fā)明提供了一種基于圖像數(shù)據(jù)增強(qiáng)的圖像訓(xùn)練集生成方法和模型訓(xùn)練方法，其中，圖像訓(xùn)練集生成方法包括：構(gòu)造初始圖像訓(xùn)練集后訓(xùn)練初始目標(biāo)檢測(cè)模型；對(duì)原始訓(xùn)練樣本圖像進(jìn)行重檢測(cè)后輸出檢測(cè)輸出圖像；比較各檢測(cè)輸出圖像上形成的檢測(cè)框和原始訓(xùn)練樣本圖像上的標(biāo)注框，將檢測(cè)框框選的錯(cuò)誤目標(biāo)圖像塊裁切下；在選中的原始訓(xùn)練樣本圖像上裁切下剔除圖像塊，選取錯(cuò)誤目標(biāo)圖像塊補(bǔ)償在剔除圖像塊的原始位置處形成負(fù)樣本圖像；混合進(jìn)入初始圖像訓(xùn)練集中形成圖像訓(xùn)練集。本發(fā)明解決了利用現(xiàn)有技術(shù)中的數(shù)據(jù)增強(qiáng)方法擴(kuò)充的圖樣訓(xùn)練集，能夠提升訓(xùn)練出的目標(biāo)檢測(cè)模型的泛化能力，但無(wú)法提高目標(biāo)檢測(cè)模型對(duì)特定的目標(biāo)對(duì)象的識(shí)別準(zhǔn)確率的問(wèn)題。