[发明专利]一种图像背景替换方法、系统、电子设备及存储介质

权利要求书

1.一种图像背景替换方法，其特征在于，包括：

获取第一图像中每个像素点的前景预测概率；

选取所述前景预测概率接近特征值的点作为不确定点；

将所述不确定点的不同尺度的特征图进行融合，得到边界清晰的前景图和α遮罩；

使用待替换的背景图像与所述前景图和α遮罩进行合成，得到新背景下的第二图像。

2.根据权利要求1所述的图像背景替换方法，其特征在于，所述获取第一图像中每个像素点的前景预测概率包括：

获取第一图像；

将所述第一图像输入卷积神经网络得到该图像的特征图；

将所述特征图输入金字塔池化模块，以获得每个像素点的前景预测概率。

3.根据权利要求2所述的图像背景替换方法，其特征在于，所述将所述特征图输入金字塔池化模块包括：

构建多层的特征金字塔，

将所述特征图输入各层金字塔进行不同尺度的池化；

进行上采样、全连接层、激活函数处理后，得到每个像素点的前景预测概率。

4.根据权利要求1所述的图像背景替换方法，其特征在于，所述选取所述前景预测概率接近特征值的点作为不确定点还包括：

反卷积处理，得到粗糙分割的特征图。

5.根据权利要求4所述的图像背景替换方法，其特征在于，所述选取所述前景预测概率接近特征值的点作为不确定点包括：

采用双线性插值对所述粗糙分割的特征图的边界进行上采样；

获取所述边界的像素点的前景预测概率；

选择概率接近特征值的点作为不确定点。

6.根据权利要求5所述的图像背景替换方法，其特征在于，所述前景预测概率来于抠图算法的公式：

I＝αF+(1-α)B

其中I为图像，F为I的前景，α为像素的前景预测概率，B为I的背景。

7.根据权利要求6所述的图像背景替换方法，其特征在于，所述选取所述前景预测概率接近特征值的点作为不确定点中所述特征值为0.5。

8.根据权利要求3所述的图像背景替换方法，其特征在于，所述将所述不确定点的不同尺度的特征图进行融合，得到边界清晰的前景图和α遮罩包括：

将所述特征图输入各层金字塔进行不同尺度的池化，得到不同尺度的特征图；

将所述不确定点的不同尺度的特征图进行融合，得到边界清晰的特征图。

9.根据权利要求8所述的图像背景替换方法，其特征在于，所述将所述不确定点的不同尺度的特征图进行融合还包括：

解码，分别得到边界清晰的前景图和α遮罩。

10.根据权利要求9所述的图像背景替换方法，其特征在于，所述解码得到边界清晰的前景图包括：

双线性差值上采样、卷积、BN、ReLU处理，得到预测的特征图；

对所述预测的特征图和所述边界清晰的特征图进行双线性差值上采样、卷积、BN、ReLU、镜面Padding、卷积，得到边界清晰的前景图。

11.根据权利要求9所述的图像背景替换方法，其特征在于，所述解码得到边界清晰的α遮罩包括：

双线性差值上采样、卷积、BN、ReLU处理，进行解码；

镜面Padding、卷积、Tanh处理，得到边界清晰的α遮罩。

12.根据权利要求1所述的图像背景替换方法，其特征在于，所述使用待替换的背景图像与所述前景图和α遮罩进行合成包括：

将α遮罩与前景图进行处理得到第二图像的前景部分；

将α遮罩与待替换的背景图像进行处理得到第二图像的背景部分；

将所述前景部分与所述背景部分进行叠加得到第二图像。