[发明专利]一种图像提取方法、装置及电子设备

说明书

本发明实施例提供了一种图像提取方法、装置及电子设备，方法包括：获取连续的视频帧；按预设周期，在预设周期内的全部视频帧中，选取基准视频帧；将基准视频帧输入图像语义分割网络，并得到与基准视频帧对应的三值图，三值图中至少包括预设目标的轮廓与背景之间的过渡区域；通过三值图，确定出预设周期内的每一个视频帧中，与三值图中的过渡区域对应的图像区域，并针对每一个视频帧，确定出图像区域是否属于该视频帧中的目标图像，当图像区域属于该视频帧中的目标图像时，将图像区域加入目标图像中；提取每一个视频帧中的目标图像。从而可以在不需要人工确定预设目标轮廓位置的情况下，能够提取出轮廓清晰，具有完整细节的目标图像。

技术领域

本发明涉及图像处理技术领域，特别是涉及一种图像提取方法、装置及电子设备。

背景技术

随着各类视频采集技术的不断更新，对于视频图像的处理越来越得到人们的关注。各类视频图像梳理方法中，从视频图像中提取各类特定的目标的图像的方法，尤其是从实时采集的视频图像中，实时提取特定的目标图像的技术手段，由于存在众多的应用领域，从而更加值得进行研究。

例如，在视频直播领域中，可以从实时采集的视频图像中，提取用户感兴趣的特定目标图像，如，从视频图像中，提取仅含有主播的人物图像，并将该图像进行渲染或其他处理，并实时的将经过处理的含有主播的人物图像，发送至播放终端，使观看直播的用户具有更好的观看体验。

由于现有的图像提取技术在提取特定目标图像时，当目标具有比较复杂的背景，或者目标的轮廓较为复杂时，一般需要人工对所要提取的目标的边缘轮廓位置进行标记或选择，从而才能够较为准确的，提取具有完整细节的目标图像。然而，当需要在获取的视频图像中，提取每一帧视频图像中的目标图像时，显然不能通过人工的方式对所要提取的目标的边缘位置进行标记或选择，所以，可能会造成所提取目标图像边缘不清晰，或者遗漏该目标边缘所具有的细节特征的问题。

发明内容

本发明实施例的目的在于提供一种图像提取方法、装置及电子设备，以实现能够从实时获取的视频图像中，提取出边缘清晰，并且边缘具有完整细节特征的目标图像。具体技术方案如下：

本发明实施例提供了一种图像提取方法，包括：

获取连续的视频帧，所述连续的视频帧中的每一个视频帧中均含有预设目标的目标图像；

按预设周期，在所述预设周期内的全部视频帧中，选取基准视频帧；

将所述基准视频帧输入图像语义分割网络，得到与所述基准视频帧对应的三值图，所述三值图中至少包括所述预设目标的轮廓与背景之间的过渡区域，所述图像语义分割网络为使用与所述预设目标的类型相同的样本进行训练后的深度学习模型；

通过所述三值图，确定出所述预设周期内的每一个视频帧中，与所述三值图中的过渡区域对应的图像区域，并针对每一个视频帧，确定出所述图像区域是否属于该视频帧中的目标图像，当所述图像区域属于该视频帧中的目标图像时，将所述图像区域加入所述目标图像中；

针对所述预设周期内的每一个视频帧，提取该视频帧中的所述目标图像。

可选的，所述按预设周期，在所述预设周期内的全部视频帧中，选取基准视频帧，包括：

按所述预设周期，将所述预设周期内的首个视频帧，作为基准视频帧。

可选的，所述三值图包括：第一区域，第二区域和所述过渡区域，所述第一区域为已确定的为所述预设目标的目标图像，所述第二区域为已确定的非所述预设目标的背景图像，所述过渡区域为所述预设目标的轮廓与背景之间的待定图像。

可选的，所述将所述基准视频帧输入图像语义分割网络，得到与所述基准视频帧对应的三值图，包括：

将所述基准视频帧输入图像语义分割网络，通过图像语义分割网络确定所述基准视频帧中每一个像素点的特征值；