[发明专利]图像处理方法、装置、存储介质及电子设备

说明书

本申请公开了一种图像处理方法、装置、存储介质及电子设备。该方法包括：获取需要进行人像分割的待分割图像；获取预训练的人像分割模型，所述预训练的人像分割模型包括：编码模块，特征金字塔模块和解码模块；调用所述编码模块对所述待分割图像进行编码处理，得到第一特征图集合；将所述第一特征图集合输入所述特征金字塔模块，得到第二特征图集合；调用所述解码模块对所述第二特征图集合进行解码处理，得到人像分割掩膜；根据所述人像分割掩膜，从所述待分割图像中分割出人像。本申请可以提高人像分割模型预测的精度，从而可更好的从图像中分割出人像。

技术领域

本申请属于电子技术领域，尤其涉及一种图像处理方法、装置、存储介质及电子设备。

背景技术

人像分割，是一种将图像中的人像从背景中分离出来的技术。人像分割是计算机视觉领域的基础课题之一，在学术界与工业界都受到了广泛的重视。

相关技术中，当需要进行人像分割处理时，通常会将图像输入预先训练好的人像分割模型中，以使该人像分割模型对图像进行人像分割处理。然而，相关技术中，人像分割模型预测的精度较低。

发明内容

本申请实施例提供一种图像处理方法、装置、存储介质及电子设备，可以提高人像分割模型预测的精度，从而可更好的从图像中分割出人像。

本申请实施例提供一种图像处理方法，包括：

获取需要进行人像分割的待分割图像；

获取预训练的人像分割模型，所述预训练的人像分割模型包括：编码模块，特征金字塔模块和解码模块；

调用所述编码模块对所述待分割图像进行编码处理，得到第一特征图集合；

将所述第一特征图集合输入所述特征金字塔模块中，得到第二特征图集合；

调用所述解码模块对所述第二特征图集合进行解码处理，得到人像分割掩膜；

根据所述人像分割掩膜，从所述待分割图像中分割出人像。

本申请实施例提供一种图像处理装置，包括：

第一获取模块，用于获取需要进行人像分割的待分割图像；

第二获取模块，用于获取预训练的人像分割模型，所述预训练的人像分割模型包括：编码模块，特征金字塔模块和解码模块；

第一调用模块，用于调用所述编码模块对所述待分割图像进行编码处理，得到第一特征图集合；

输入模块，用于将所述第一特征图集合输入所述特征金字塔模块中，得到第二特征图集合；

第二调用模块，用于调用所述解码模块对所述第二特征图集合进行解码处理，得到人像分割掩膜；

分割模块，用于根据所述人像分割掩膜，从所述待分割图像中分割出人像。

本申请实施例提供一种存储介质，其上存储有计算机程序，当所述计算机程序在计算机上执行时，使得所述计算机执行本申请实施例提供的图像处理方法中的流程。

本申请实施例还提供一种电子设备，包括存储器，处理器，所述处理器通过调用所述存储器中存储的计算机程序，用于执行本申请实施例提供的图像处理方法中的流程。

本申请实施例中，可将对图像进行编码处理得到的第一特征图集合输入特征金字塔模块，得到第二特征图集合；并调用解码模块对第二特征图集合进行解码处理，得到人像分割掩膜，从而可充分利用第一特征图集合，以更好的提取语义信息，进而可提高人像分割模型预测的精度，进而可更好的从图像中分割出人像。

附图说明

下面结合附图，通过对本申请的具体实施方式详细描述，将使本申请的技术方案及其有益效果显而易见。