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

基本信息

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

摘要

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