[发明专利]精分割天空区域的方法及系统、图像换天的方法及系统

说明书

本发明公开一种精分割天空区域的方法及系统、图像换天的方法及系统，其中精分割天空区域的方法包括以下步骤：获取待分割图像中每个像素点属于天空的置信度；根据所述置信度生成与所述待分割图像相对应的第一掩膜图像和第一张三分图；对所述第一掩膜图像进行形态学处理，生成第二张三分图；将所述第一张三分图和所述第二张三分图进行合成，生成第三张三分图，基于第三张三分图对待分割图像进行分割，输出分割结果。本发明能够对天空与非天空的边界进行精细抠图，实现对天空区域的精分割。

技术领域

本发明涉及图像处理领域，尤其涉及一种精分割天空区域的方法及系统、图像换天的方法及系统。

背景技术

现今天空区域识别技术通常根据天空场景颜色等特征、以及预设的阈值对天空区域进行提取，天空分割一般采用边缘检测算法检测图像中天空区域边界，或采用基于深度学习的语义分割模型将图像分割为天空区域和非天空区域。但上述方法所提取的天空区域边界都比较粗糙，并且上述方法都无法处理树、建筑等非天空区域缝隙间的天空，使提取的天空区域/非天空区域不完整，应用所提取的天空区域/非天空区域合成图像时易出现瑕疵。

故需要对现有技术做进一步改进。

发明内容

本发明针对现有技术中的缺点，提供了一种精分割天空区域的方法及系统、还提供一种图像换天的方法及系统。

为了解决上述技术问题，本发明通过下述技术方案得以解决：

一种精分割天空区域的方法，包括以下步骤：

获取待分割图像中每个像素点属于天空的置信度；

根据所述置信度生成与所述待分割图像相对应的第一掩膜图像和第一张三分图；

对所述第一掩膜图像进行形态学处理，生成第二张三分图；

将所述第一张三分图和所述第二张三分图进行合成，生成第三张三分图，基于第三张三分图对待分割图像进行分割，输出分割结果。

作为一种可实施方式，对所述第一掩膜图像进行形态学处理，生成第二张三分图的具体步骤为：

对所述第一掩膜图像进行腐蚀操作，获得第二掩膜图像，对所述第一掩膜图像进行膨胀操作，获得第三掩膜图像，将所述第二掩膜图像和所述第三掩膜图像进行合成，生成第二张三分图。

作为一种可实施方式，将所述第一张三分图和所述第二张三分图进行合成，生成第三张三分图，基于第三张三分图对待分割图像进行分割，输出分割结果具体步骤为：

所述第一张三分图、所述第二张三分图和所述第三张三分图均包括非天空的前景区域、天空背景区域和待抠图区域；将所述第一张三分图和所述第二张三分图中非天空的前景区域的交集作为第三张三分图的非天空的前景区域，将所述第一张三分图和所述第二张三分图中天空背景区域的交集作为第三张三分图的天空背景区域，其余部分均作为第三张三分图的待抠图区域；

采用图像抠取算法，基于所述第三张三分图对待分割图像进行分割，输出分割结果。

本发明还提出一种精分割天空区域的系统，包括：

置信度获取模块，用于获取待分割图像中每个像素点属于天空的置信度；

预分割模块，用于根据所述置信度生成与所述待分割图像相对应的第一掩膜图像和第一张三分图；

形态学处理模块，用于对所述第一掩膜图像进行形态学处理，生成第二张三分图；

精分割模块，用于将所述第一张三分图和所述第二张三分图进行合成，生成第三张三分图，基于第三张三分图对待分割图像进行分割，输出分割结果。

作为一种可实施方式，所述形态学处理模块被配置为：