[发明专利]一种图像处理方法和装置

权利要求书

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

接收图像处理请求，所述图像处理请求指示需要处理的图像、以及需要替换的元素类型；

获取与所述元素类型对应的语义分割模型，所述语义分割模型由深度神经网络训练而成；

根据所述语义分割模型，对所述图像中每一像素属于所述元素类型的概率进行预测，得到初始概率图；

基于条件随机场对所述初始概率图进行优化，得到分割效果图；

根据所述分割效果图将所述图像与预设元素素材进行融合，得到处理后图像；

其中所述根据所述分割效果图将所述图像与预设元素素材进行融合，得到处理后图像的步骤包括：

按照预设策略获取可替换的元素素材；

通过融合算法将所述分割效果图中的第一颜色部分与获取到的元素素材相结合，得到第一结合图；

通过融合算法将所述分割效果图中的第二颜色部分与所述图像相结合，得到第二结合图；

将第一结合图和第二结合图进行合成，得到处理后图像。

2.根据权利要求1所述的方法，其特征在于，所述根据所述语义分割模型，对所述图像中每一像素属于所述元素类型的概率进行预测，得到初始概率图，包括：

将所述图像导入所述语义分割模型，以预测所述图像中每一像素属于所述元素类型的概率；

根据所述概率对相应像素在预设蒙版上的颜色进行设置，得到初始概率图。

3.根据权利要求2所述的方法，其特征在于，所述根据所述概率对相应像素在预设蒙版上的颜色进行设置，得到初始概率图，包括：

确定所述概率是否大于预设阈值；

若是，则将相应像素在预设蒙版上的颜色设置为第一颜色；

若否，则将相应像素在预设蒙版上的颜色设置为第二颜色；

在确定所述图像中所有像素在预设蒙版上的颜色均设置完毕后，输出设置颜色后的预设蒙版，得到初始概率图。

4.根据权利要求1所述的方法，其特征在于，所述基于条件随机场对所述初始概率图进行优化，得到分割效果图，包括：

将所述初始概率图中的像素映射至条件随机场中的节点；

确定节点之间的边约束的相似性；

根据边约束的相似性对所述初始概率图中像素的分割结果进行调整，得到分割效果图。

5.根据权利要求1所述的方法，其特征在于，所述按照预设策略获取可替换的元素素材，包括：

获取候选图像，对所述候选图像进行随机截取，并将截取到的图像作为可替换的元素素材；或者，

接收用户触发的素材选择指令，根据所述素材选择指令从素材库获取相应的素材，作为可替换的元素素材。

6.根据权利要求1至4任一项所述的方法，其特征在于，所述根据所述分割效果图将所述图像与预设元素素材进行融合，得到处理后图像之前，还包括：

对所述分割效果图进行外观模型算法和/或图像形态学操作处理，得到处理后分割效果图；

所述根据所述分割效果图将所述图像与预设元素素材进行融合，得到处理后图像，包括：根据处理后分割效果图，将所述图像与预设元素素材进行融合，得到处理后图像。

7.根据权利要求1至4任一项所述的方法，其特征在于，所述获取与所述元素类型对应的语义分割模型之前，还包括：

获取包含有所述元素类型的训练数据；

根据所述训练数据，利用深度神经网络对预设的语义分割初始模型进行训练，得到所述元素类型对应的语义分割模型。