一種交互式小樣本語義分割的訓練方法

基本信息

申請?zhí)?/td> CN202111223958.3 申請日 -
公開(公告)號 CN113936178A 公開(公告)日 2022-01-14
申請公布號 CN113936178A 申請公布日 2022-01-14
分類號 G06V10/764(2022.01)I;G06V10/774(2022.01)I;G06V10/778(2022.01)I;G06K9/62(2022.01)I 分類 計算;推算;計數;
發(fā)明人 楊培文;張成英;梁惠瑩;于振東;張遼 申請(專利權)人 深圳市玻爾智造科技有限公司
代理機構 杭州九洲專利事務所有限公司 代理人 張羽振
地址 518110廣東省深圳市龍華區(qū)觀瀾街道大富社區(qū)大富工業(yè)區(qū)20號硅谷動力智能終端產業(yè)園A5棟101
法律狀態(tài) -

摘要

摘要 本發(fā)明涉及一種交互式小樣本語義分割的訓練方法,包括步驟:在原始圖片上繪制不同類別的點,用點標注待學習目標;將標注出待學習目標的圖片輸入預訓練模型,進行特征提??;將標注點投影到特征圖上得到特征向量和類別標簽,使用邏輯回歸模型對原始標注點進行分類和預測;將單張圖片作為小樣本,學習小樣本不同類別標注點的特征。本發(fā)明的有益效果是:通過對樣本圖片進行交互式的標注學習目標、特征提取與訓練、過檢和漏檢修復等操作,可以達到僅使用一張圖也可以訓練出效果不錯的模型效果,解決了當前目標檢測常見方法中標注數據工作量大和使用小樣本數據訓練模型泛化能力弱的問題。