一种图像抠图的方法、设备和存储介质技术

本发明专利技术实施例公开了一种图像抠图的方法、设备和存储介质，该方法包括生成标识检测模型和标识抠图模型，对于用户的带有标识的图像，根据标识检测模型，将其中检测到的标识区域裁剪下来，获得第二图像；根据标识抠图模型，对第二图像进行抠图，获得标识的掩码图；新建一张与用户的带有标识的图像同样大小的像素值都为0的掩码图，将标识的掩码图替换用户带有标识的掩码图中标识区域位置的内容；将新的掩码图作为透明通道a，合在用户的带有标识的图像的rgb图上，形成带有透明通道的rgba图，获得最终的抠图结果。采用了本发明专利技术的技术方案，能够提高标识抠取的精度，加大标识图像的分辨率，而且减少显卡消耗，提升推理速度。

【技术实现步骤摘要】

本专利技术涉及图像抠图，尤其涉及一种图像抠图的方法、设备和存储介质。

技术介绍

1、目前通用的标识抠图模型抠物的种类很多，但是对于一些特定类型的目标，如标识(logo)，抠取精度不够，没法满足用户需求，特别是当logo较小时，logo抠取结果的精度损失严重。

2、同时常用的标识抠图模型，如u2net,模型输入的图像尺寸过小(像素为320*320)。当用户图像较大时，如果将用户图像缩小尺度太大，会损失很多原图的细节信息，导致抠图结果很多细节抠不好；但是如果改大模型的输入尺寸，训练模型对显卡的要求急剧增加。

技术实现思路

1、为了解决上述技术问题，本专利技术提出了一种图像抠图的方法、设备和存储介质。

2、本专利技术实施例提供的一种图像抠图的方法，包括以下步骤：

3、从网络中爬取带有标识的图像，对所述带有标识的图像中的标识区域进行标注，标注信息包括标识区域的位置信息和标识的类别标签，采用目标检测方法生成标识检测模型；

4、将所述带有标识的图像中的标识区域裁剪下来，再对本文档来自技高网...

【技术保护点】

1.一种图像抠图的方法，其特征在于，包括以下步骤：

2.根据权利要求1所述的图像抠图的方法，其特征在于，所述位置信息进一步包括标识区域的矩形左上坐标点位置和右下坐标点位置。

3.根据权利要求1所述的图像抠图的方法，其特征在于，所述标识检测模型采用yolov7。

4.根据权利要求3所述的图像抠图的方法，其特征在于，还包括以下步骤：

5.根据权利要求4所述的图像抠图的方法，其特征在于，还包括以下步骤：

6.根据权利要求1所述的图像抠图的方法，其特征在于，所述标识抠图模型采用的5个编码器和4个解码器均为残差U型模块，所述残差U型模块使...

【技术特征摘要】

1.一种图像抠图的方法，其特征在于，包括以下步骤：

2.根据权利要求1所述的图像抠图的方法，其特征在于，所述位置信息进一步包括标识区域的矩形左上坐标点位置和右下坐标点位置。

3.根据权利要求1所述的图像抠图的方法，其特征在于，所述标识检测模型采用yolov7。

4.根据权利要求3所述的图像抠图的方法，其特征在于，还包括以下步骤：

5.根据权利要求4所述的图像抠图的方法，其特征在于，还包括以下步骤：

6.根据权利要求1所述的图像抠图的方法，其特征在于，所述标识抠图模型采用的5个编码器和4个解码器均为残差u型模块，所述残差u型模块使用u型结构连接。

7.根据权利要求1所述的图像抠图的方法，其特征在于，所述从网络中爬取带有标识的图像，按照4:1:1分成训练集、验证集和测试集，其中训练集用于训练标识检测模型，验证集用于在训练过程...

【专利技术属性】
技术研发人员：万林，马妙聪，马成龙，陈思远，
申请(专利权)人：苏州大创科技有限公司，
类型：发明
国别省市：