[发明专利]视频人像抠图方法、系统和介质

说明书

本发明公开了视频人像抠图方法、系统和介质，其中视频人像抠图方法包括：获取预训练模型，获取第一视频；对第一视频的每帧图片进行裁剪，对每帧图片的像素在空域进行下采样，以生成第二视频；对第二视频在时域进行跳跃采帧获得采样帧；将采样帧输入预训练模型以生成alpha遮罩，将alpha遮罩作为采样帧的相邻帧的alpha遮罩；根据alpha遮罩和alpha遮罩对第二视频的所有图片帧进行对齐裁剪，以获得去除背景后的第二视频。能够大幅提升视频抠图的速度和效率，而且还能有效保持人像的边缘细节，具有突出的特点，能够满足视频直播等场景的实时抠图需求，提升视频传输的速度，能够给用户带来较好的观看体验。

技术领域

本发明涉及计算机视觉及图像处理技术领域，特别涉及一种视频人像抠图方法、系统和介质。

背景技术

人像抠图是从给定的图像或视频中提取人物，有着各种各样的应用，如短视频背景替换、直播背景更换、电影场景背景后期设计，针对视频的抠图方法通常有基于绿色荧幕来实时获得高质量的人像提取，但是绿幕抠图的场景单一，对背景要求较高，抠图处理的速度慢，且图像处理的成本较高。

发明内容

本发明旨在至少解决现有技术中存在的技术问题之一。为此，本发明提出一种视频人像抠图方法，能够实现人像高质量扣取并且拥有较快的人像提取处理速度，降低图像处理的成本，能够实时处理人像提取输出。

本发明还提出一种具有上述视频人像抠图方法的视频人像抠图系统。

本发明还提出一种计算机可读存储介质。

一方面，本实施例提供了一种视频人像抠图方法，包括：

获取预训练模型，获取第一视频；

对所述第一视频的每帧图片进行裁剪，对每帧所述图片的像素在空域进行下采样，以生成第二视频；

对所述第二视频在时域进行跳跃采帧获得采样帧；

将所述采样帧输入所述预训练模型以生成alpha遮罩，将所述alpha遮罩作为所述采样帧的相邻帧的alpha遮罩；

根据所述采样帧的alpha遮罩和所述相邻帧的alpha遮罩对所述第二视频的所有图片帧进行对齐裁剪，以获得去除背景后的所述第二视频。

根据本发明实施例的视频人像抠图方法，至少具有如下有益效果：首先，获取预训练模型和第一视频，第一视频为包含人像数据的待处理视频，将第一视频的每一帧图片进行处理，包括对每一帧图片进行裁剪处理，可以是粗略地获取第一视频内的人像数据，再对每一帧图片在空间域进行像素下采样，以生成第二视频；对第二视频的图片帧在时域上进行跳跃采帧，可以是只采样奇数帧输入预训练模型以得到奇数帧的alpha遮罩，再将alpha遮罩作为奇数帧的相邻帧的alpha遮罩，奇数帧的相邻帧即为偶数帧，结合采样帧的alpha遮罩和相邻帧的alpha遮罩即可得到第二视频的所有帧的alpha遮罩，基于时域对第二视频进行采样还可以生成丢失的帧的alpha遮罩，根据采样帧的alpha遮罩和相邻帧的alpha遮罩对第二视频的所有图片帧进行对齐裁剪，以获得去除背景后的第二视频。本实施例的视频人像抠图方法能够实现人像高质量扣取并且拥有较快的人像提取处理速度，降低图像处理的成本，能够实时处理人像提取输出，满足基本的实时直播应用场景。

根据本发明的一些实施例，所述获取预训练模型，包括步骤：

获取包括人像数据的第一图像和与所述第一图像对应的图像遮罩；

对所述第一图像进行下采样，根据所述下采样的结果和所述人像数据获取人像遮罩；

获取所述人像数据的边缘细节，将所述边缘细节与所述人像遮罩合成以得到第二图像；

根据所述第二图像和所述图像遮罩进行学习以获得预训练模型。