一種基于語義分割的圖像trimap生成方法

基本信息

申請?zhí)?/td> CN201910677503.5 申請日 -
公開(公告)號 CN110363728A 公開(公告)日 2019-10-22
申請公布號 CN110363728A 申請公布日 2019-10-22
分類號 G06T5/00;G06T5/30;G06N3/08;G06N3/04 分類 計算;推算;計數(shù);
發(fā)明人 胡玉琛;李猛;仇文彬 申請(專利權(quán))人 上海尊溢商務(wù)信息咨詢有限公司
代理機構(gòu) 上海卓陽知識產(chǎn)權(quán)代理事務(wù)所(普通合伙) 代理人 上海尊溢商務(wù)信息咨詢有限公司
地址 200072 上海市靜安區(qū)萬榮路1188弄7、8、11號212室
法律狀態(tài) -

摘要

摘要 本發(fā)明提供了一種基于語義分割的圖像trimap生成方法,包括以下步驟:S1,獲取海量原始圖像數(shù)據(jù)與其標(biāo)注數(shù)據(jù);S2,對數(shù)據(jù)進行預(yù)處理,提取為RGB色空間的三維特征數(shù)據(jù);S3,將三維特征數(shù)據(jù)輸入深度神經(jīng)網(wǎng)絡(luò)模型進行訓(xùn)練,得到權(quán)重;S4,利用訓(xùn)練得到的模型權(quán)重對新圖像進行預(yù)測,并使用條件隨機場對結(jié)果進行優(yōu)化,得到分割掩碼圖;S5,對步驟S4中產(chǎn)生的分割掩碼圖進行圖像形態(tài)學(xué)處理,得到精確的原圖trimap圖。本發(fā)明的方法適用于任何前景物體,并且不局限于簡單背景與復(fù)雜背景,在抗干擾與魯棒性上非常優(yōu)秀,產(chǎn)生的trimap精度高、速度快。