抠图方法、装置、电子设备以及存储介质制造方法及图纸

本申请公开了一种抠图方法、装置、电子设备以及存储介质，涉及计算机技术领域。该方法包括：获取待抠图图像；将待抠图图像输入目标检测模型，获得目标检测模型输出的从待抠图图像中确定的待抠图子图像，以及待抠图子图像对应的类别置信度；若类别置信度大于或等于置信度阈值，则将待抠图子图像输入已训练的抠图模型，获得已训练的抠图模型输出的与待抠图子图像对应的目标掩膜图像；基于待抠图子图像以及目标掩膜图像，获得抠图图像。本申请通过在抠图前使用目标检测模型获取待抠图子图像，并且利用已训练的抠图模型对待抠图子图像进行抠图，提高了抠图的质量，也提高了用户的体验感。也提高了用户的体验感。也提高了用户的体验感。

【技术实现步骤摘要】
抠图方法、装置、电子设备以及存储介质


[0001]本申请涉及计算机
，更具体地，涉及一种抠图方法、装置、电子设备以及存储介质。

技术介绍

[0002]随着科学技术的进步，图像处理技术得到了迅猛的发展。相关技术中，提出了通过分离图像或者视频中的前景与背景画面，实现高精度的前景提取与虚拟背景替换工作的数字图像抠图技术。然而相关技术中由于开源的抠图算法模型体量庞大且通用性和鲁棒性欠佳，利用开源的抠图算法抠图存在用户体验感不佳的问题。

技术实现思路

[0003]鉴于上述问题，本申请提出了一种抠图方法、装置、电子设备以及存储介质，可以通过在抠图前使用目标检测模型获取待抠图子图像，并且利用已训练的抠图模型对待抠图子图像进行抠图，提高了抠图的质量，也提高了用户的体验感。
[0004]第一方面，本申请实施例提供了一种抠图方法，所述方法包括：获取待抠图图像；将所述待抠图图像输入目标检测模型，获得所述目标检测模型输出的从所述待抠图图像中确定的待抠图子图像，以及所述待抠图子图像对应的类别置信度；若所述类别置信度大于或等于置信度阈值，则将本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种抠图方法，其特征在于，所述方法包括：获取待抠图图像；将所述待抠图图像输入目标检测模型，获得所述目标检测模型输出的从所述待抠图图像中确定的待抠图子图像，以及所述待抠图子图像对应的类别置信度；若所述类别置信度大于或等于置信度阈值，则将所述待抠图子图像输入已训练的抠图模型，获得所述已训练的抠图模型输出的与所述待抠图子图像对应的目标掩膜图像；基于所述待抠图子图像以及所述目标掩膜图像，获得抠图图像。2.根据权利要求1所述的方法，其特征在于，在所述基于所述待抠图子图像以及所述目标掩膜图像，获得抠图图像之后，还包括：若所述抠图图像的短边的像素小于或等于第一像素阈值，则基于第一超分辨率放大所述抠图图像，获得目标抠图图像；或者若所述抠图图像的短边的像素小于或等于第二像素阈值且大于所述第一像素阈值，则基于第二超分辨率放大所述抠图图像，获得目标抠图图像，其中，所述第一像素阈值小于所述第二像素阈值，所述第一超分辨率大于所述第二超分辨率；或者若所述抠图图像的短边的像素大于所述第二像素阈值，则确定所述抠图图像作为目标抠图图像。3.根据权利要求1所述的方法，其特征在于，所述待抠图图像中包括多个主体，所述将所述待抠图图像输入目标检测模型，获得所述目标检测模型输出的从所述待抠图图像中确定的待抠图子图像，以及所述待抠图子图像对应的类别置信度，包括：获取抠图指令；基于所述抠图指令从所述多个主体中确定目标主体；将所述待抠图图像输入所述目标检测模型，获得所述目标检测模型输出的与所述目标主体对应的从所述待抠图图像中确定的待抠图子图像，以及所述待抠图子图像对应的类别置信度。4.根据权利要求1所述的方法，其特征在于，所述将所述待抠图图像输入目标检测模型，获得所述目标检测模型输出的从所述待抠图图像中确定的待抠图子图像，以及所述待抠图子图像对应的类别置信度，包括：将所述待抠图图像输入所述目标检测模型，获得所述目标检测模型输出的从所述待抠图图像中确定的目标检测框，以及所述目标检测框对应的类别置信度；基于所述目标检测框以及所述目标检测框对应的类别置信度，获得所述待抠图子图像以及所述待抠图子图像对应的类别置信度。5.根据权利要求1所述的方法，其特征在于，在所述若所述类别置信度大于或等于置信度阈值，则将所述待抠图子图像输入已训练的抠图模型，获得所述已训练的抠图模型输出的目标掩膜图像之前，还包括：获取样本数据集，其中，所述样本数据集中包括多张图像以及各所述图像对应的掩膜图像标签；基于所述多张图像以及各所述图像对应的掩膜图像标签训练初始抠图模型，获得所述已训练的抠图模型。6.根据权利要求5所述的方法，其特征在于，所述初始抠图模型中包括至少三个编码层
以及对应所述至少三个编码层...

【专利技术属性】
技术研发人员：王凡祎，张严浩，冯天鹏，李森，张济智，朱莉，
申请(专利权)人：上海瑾盛通信科技有限公司，
类型：发明
国别省市：