[发明专利]一种图像分割方法、装置、电子设备及存储介质

说明书

本发明公开了一种图像分割方法、装置、电子设备及存储介质，在对待分割的第一图像进行图像分割时，基于图像分割模型的第一网络分支对第一图像进行特征提取得到第一特征图，结合邻近帧第二图像对应的第二特征图，得到拼接后的第三特征图，其中，第二特征图是基于图像分割模型的第一网络分支和第二网络分支对第二图像进行特征提取得到的。最后基于图像分割模型的第三网络分支，对第三特征图进行特征提取，得到图像分割结果。减少了图像分割模型的第二网络分支对第一图像进行特征提取的过程，从而减少了图像分割的计算量，并且结合第一特征图和第二特征图，进行最终的图像分割结果的确定，避免了分割效果较差，帧间容易出现闪烁显示的问题。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种图像分割方法、装置、电子设备及存储介质。

背景技术

直播平台中主播使用手机等设备进行直播时，在一些场景下希望进行图像分割，例如将人像扣除，然后替换准备好的背景图片。由于手机的硬件性能有限，在使用图像分割模型是会出现帧率下降等不良影响。目前直播平台普遍采用以下方案：根据手机类型设计不同计算量的模型，高档机计算量在300M次浮点运算，中档机120M次浮点运算，低档机次80M次浮点运算。

在实现图像分割的过程中，发现现有技术中至少存在如下问题：中档机或低档机因为计算量限制导致分割效果较差，图像分割的效果较差，并且因为计算量限制帧间容易出现闪烁显示，例如衣服的某一小块区域前一帧检测为人像，后一帧检测为背景，该情形会有肉眼可见的闪烁。

发明内容

本发明实施例提供了一种图像分割方法、装置、电子设备及存储介质，用以解决相关技术的图像分割方案因为计算量限制导致分割效果较差，图像分割的效果较差，并且容易出现闪烁显示的问题。

本发明实施例提供了一种图像分割方法，所述方法包括：

获取待分割的第一图像及所述第一图像的邻近帧第二图像；将所述第一图像输入预先训练完成的图像分割模型，基于所述图像分割模型的第一网络分支对所述第一图像进行特征提取得到第一特征图；

获取所述第二图像对应的第二特征图，其中，所述第二特征图是将所述第二图像输入所述图像分割模型，基于所述图像分割模型的第一网络分支和第二网络分支对所述第二图像进行特征提取得到的；

对所述第一特征图和第二特征图进行拼接，得到第三特征图，基于所述图像分割模型的第三网络分支，对所述第三特征图进行特征提取，得到图像分割结果。

另一方面，本发明实施例提供了一种图像分割装置，所述装置包括：

第一获取模块，用于获取待分割的第一图像及所述第一图像的邻近帧第二图像；将所述第一图像输入预先训练完成的图像分割模型，基于所述图像分割模型的第一网络分支对所述第一图像进行特征提取得到第一特征图；

第二获取模块，用于获取所述第二图像对应的第二特征图，其中，所述第二特征图是将所述第二图像输入所述图像分割模型，基于所述图像分割模型的第一网络分支和第二网络分支对所述第二图像进行特征提取得到的；

第一分割模块，用于对所述第一特征图和第二特征图进行拼接，得到第三特征图，基于所述图像分割模型的第三网络分支，对所述第三特征图进行特征提取，得到图像分割结果。

再一方面，本发明实施例提供了一种电子设备，包括处理器、通信接口、存储器和通信总线，其中，处理器，通信接口，存储器通过通信总线完成相互间的通信；

存储器，用于存放计算机程序；

处理器，用于执行存储器上所存放的程序时，实现上述任一项所述的方法步骤。

再一方面，本发明实施例提供了一种计算机可读存储介质，所述计算机可读存储介质内存储有计算机程序，所述计算机程序被处理器执行时实现上述任一项所述的方法步骤。