首页 专利交易 科技果 科技人才 科技服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

一种电商产品抠图方法 

申请/专利权人:焦点科技股份有限公司

申请日:2024-04-17

公开(公告)日:2024-06-28

公开(公告)号:CN118261933A

主分类号:G06T7/194

分类号:G06T7/194;G06T5/50;G06T5/30;G06N3/0455;G06N3/0464;G06N3/0895

优先权:

专利状态码:在审-公开

法律状态:2024.06.28#公开

摘要:本发明公开了一种电商产品抠图方法,其特征在于利用高精度图像二类分割模型对电商产品进行显著性检测,预测出产品的前景和背景,基于前景和背景预测结果,采用腐蚀和膨胀操作,将二类分割图转化为包含前景、背景和边界的三分图;将产品的原始图像以及生成的三分图作为输入,再基于三分图训练改进的transformer结构抠图模型。本发明通过改进transformer结构,使用基于窗口的注意力机制、基于窗口偏移的注意力机制和相对位置编码,能更好的融合局部和全局特征信息,并将三分图的全局表征融入原始图像窗口区域的注意力计算中,得到精确的产品抠图结果,有效解决了由于忽略三分图先验信息而导致抠图蒙版预测不准确的问题。

主权项:1.一种电商产品抠图方法,其特征在于,包括以下步骤:步骤1:准备用于训练二类分割模型的第一数据集,所述二类分割模型基于transformer结构,用于对商品图中的显著性物体与背景进行分割,所述数据集包括产品的RGB图和第一Mask图;对第一Mask图进行阈值设置,用以筛选为第二Mask图,基于第二Mask图使用腐蚀膨胀处理,得到三分图;步骤2:准备用于训练抠图模型的第二数据集,所述抠图模型基于transformer结构,所述第二数据集包括产品的RGB图和Alpha标签,将所述RGB图与步骤1中获得的三分图输入抠图模型,使用基于窗口的注意力机制、基于窗口偏移的注意力机制和相对位置编码改进transformer结构,针对窗口的尺寸进行寻优训练,所述窗口大小为7,计算窗口内像素的注意力权重;将三分图输入基于窗口的注意力机制中,用以对三分图的全局表征与RGB图的特征进行信息融合,最小化预测Alpha图与所述Alpha标签之间的损失;将预测的Alpha图与原RGB图进行通道合并得到最终抠图结果。

全文数据:

权利要求:

百度查询: 焦点科技股份有限公司 一种电商产品抠图方法

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。