一种基于语义分割的图像trimap生成方法

基本信息

申请号 CN201910677503.5 申请日 -
公开(公告)号 CN110363728A 公开(公告)日 2019-10-22
申请公布号 CN110363728A 申请公布日 2019-10-22
分类号 G06T5/00;G06T5/30;G06N3/08;G06N3/04 分类 计算;推算;计数;
发明人 胡玉琛;李猛;仇文彬 申请(专利权)人 上海尊溢商务信息咨询有限公司
代理机构 上海卓阳知识产权代理事务所(普通合伙) 代理人 上海尊溢商务信息咨询有限公司
地址 200072 上海市静安区万荣路1188弄7、8、11号212室
法律状态 -

摘要

摘要 本发明提供了一种基于语义分割的图像trimap生成方法,包括以下步骤:S1,获取海量原始图像数据与其标注数据;S2,对数据进行预处理,提取为RGB色空间的三维特征数据;S3,将三维特征数据输入深度神经网络模型进行训练,得到权重;S4,利用训练得到的模型权重对新图像进行预测,并使用条件随机场对结果进行优化,得到分割掩码图;S5,对步骤S4中产生的分割掩码图进行图像形态学处理,得到精确的原图trimap图。本发明的方法适用于任何前景物体,并且不局限于简单背景与复杂背景,在抗干扰与鲁棒性上非常优秀,产生的trimap精度高、速度快。