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

权利要求书

1.一种图像处理方法，其特征在于，包括：

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

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

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

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

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

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

2.根据权利要求1所述的图像处理方法，其特征在于，所述解码模块包括第一子模块和第二子模块，所述调用所述解码模块对所述第二特征图集合进行解码处理，得到人像分割掩膜，包括：

调用所述第一子模块将所述第二特征图集合中的特征图的尺寸调整至预设尺寸，得到调整后的特征图；

调用所述第二子模块对所述调整后的特征图进行融合、卷积及采样处理，得到人像分割掩膜。

3.根据权利要求2所述的图像处理方法，其特征在于，所述第一子模块包括第一网络块、第二网络块和第三网络块，所述调用所述第一子模块将所述第二特征图集合中的特征图的尺寸调整至预设尺寸，得到调整后的特征图，包括：

调用所述第一网络块、所述第二网络块和/或所述第三网络块将所述第二特征图集合中的特征图调整至预设尺寸，得到调整后的特征图。

4.根据权利要求3所述的图像处理方法，其特征在于，所述第一网络块包括依次连接的卷积层、归一化层和激活层；

所述第二网络块包括依次连接的卷积层、归一化层、激活层和上采样层；

所述第三网络块包括依次连接的卷积层、归一化层、激活层和上采样层，所述第三网络块的输入通道数与所述第三网络块的输出通道数相同。

5.根据权利要求2所述的图像处理方法，其特征在于，所述第二子模块包括第一融合层、第一卷积层和第一上采样层，所述调用所述第二子模块对所述调整后的特征图进行融合、卷积及采样处理，得到人像分割掩膜，包括：

调用所述第一融合层对所述调整后的特征图进行融合处理，得到融合后的特征图；

调用所述第一卷积层对所述融合后的特征图进行卷积处理，得到卷积后的特征图；

调用所述第一上采样层对所述卷积后的特征图进行上采样处理，得到人像分割掩膜。

6.根据权利要求1所述的图像处理方法，其特征在于，所述获取需要进行人像分割的待分割图像之前，还包括：

获取样本图像，以及所述样本图像对应的标注掩膜；

获取监督模块；

利用所述样本图像、所述样本图像对应的标注掩膜以及所述监督模块对人像分割模型进行训练。

7.根据权利要求6所述的图像处理方法，其特征在于，所述利用所述样本图像、所述样本图像对应的标注掩膜以及所述监督模块对人像分割模型进行训练，包括：

调用所述编码模块对所述样本图像进行编码处理，得到第三特征图集合；

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

调用所述解码模块对所述第四特征图集合进行解码处理，得到期望掩膜；

调用所述监督模块对所述第四特征图集合进行还原处理，得到多个监督掩膜；

根据所述期望掩膜与所述标注掩膜的差异，以及每个监督掩模与所述标注掩膜的差异，调整所述人像分割模型的参数。

8.根据权利要求7所述的图像处理方法，其特征在于，所述监督模块包括第四卷积层和第三上采样层，所述调用所述监督模块对所述第四特征图集合进行还原处理，得到多个监督掩膜，包括：

调用所述第四卷积层对所述第四特征图集合中的特征图分别进行卷积处理，得到第五特征图集合；

调用所述第三上采样层对所述第五特征图集合中的特征图分别进行上采样处理，得到多个监督掩膜。