视频天空区域调色方法、装置、电子设备及存储介质与流程

本发明涉及视频处理，具体涉及一种视频天空区域调色方法、装置、电子设备及存储介质。

背景技术：

1、在用户对于视频中天空的色彩具有更高要求的背景下，天空调色算法可以对天空区域进行调色处理，满足用户需求。但是，天空调色的前提是能够分割出完整且没有明显误分割的天空区域。对于天空分界较为模糊的画面，或者画面中存在与天空区域十分类似的区域时，例如：雪地、倒映天空的水面等区域。由于天空颜色与这些区域比较接近，导致天空分割在这些场景中极具挑战，尤其在长视频下表现逊色。

2、目前主流的天空分割模型，往往只能适用于通用场景，对于特定场景，在使用业务数据训练之后也能够有较好的表现，但是对于雪地、水面等大面积类似天空颜色的场景，会出现明显的错误分割情况，导致在渲染后，视频帧间出现渲染异常、抖动等情况，继而影响视频美观和观看者的观看体验。

技术实现思路

1、为了解决现有技术中存在的上述问题，本申请实施方式提供了一种视频天空区域调色方法、装置、电子设备及存储介质，可以通过天空分割的历史信息对之后图像的分割情况进行调整，在不改变模型参数的前提下，处理误分割现象，解决了模型输出在视频帧间渲染异常、抖动的问题，提升视频美观和观看者的观看体验。

2、第一方面，本申请的实施方式提供了一种视频天空区域调色方法，该方法包括：

3、对视频数据进行分帧处理，得到n张视频帧图像；

4、对n张视频帧图像中的第一张视频帧图像进行天空区域识别，得到初始二值图；

5、根据第一张视频帧图像的尺寸信息，确定第一阈值；

6、根据初始二值图和第一阈值，对n-1张视频帧图像进行n-1次天空区域识别处理，得到n-1张最终二值图，其中，n-1张视频帧图像为n张视频帧图像中除去第一张视频帧图像后剩下的n-1张图像；

7、根据初始二值图和n-1张最终二值图，对n张视频帧图像进行分天空区域调色，得到调色后的视频数据。

8、第二方面，本申请的实施方式提供了一种视频天空区域调色装置，包括：

9、分帧模块，用于对视频数据进行分帧处理，得到n张视频帧图像；

10、处理模块，用于对n张视频帧图像中的第一张视频帧图像进行天空区域识别，得到初始二值图，根据第一张视频帧图像的尺寸信息，确定第一阈值，并根据初始二值图和第一阈值，对n-1张视频帧图像进行n-1次天空区域识别处理，得到n-1张最终二值图，其中，n-1张视频帧图像为n张视频帧图像中除去第一张视频帧图像后剩下的n-1张图像；

11、调色模块，用于根据初始二值图和n-1张最终二值图，对n张视频帧图像进行分天空区域调色，得到调色后的视频数据。

12、第三方面，本申请实施方式提供一种电子设备，包括：处理器，处理器与存储器相连，存储器用于存储计算机程序，处理器用于执行存储器中存储的计算机程序，以使得电子设备执行如第一方面的方法。

13、第四方面，本申请实施方式提供一种计算机可读存储介质，计算机可读存储介质存储有计算机程序，计算机程序使得计算机执行如第一方面的方法。

14、第五方面，本申请实施方式提供一种计算机程序产品，计算机程序产品包括存储了计算机程序的非瞬时性计算机可读存储介质，计算机可操作来使计算机执行如第一方面的方法。

15、实施本申请实施方式，具有如下有益效果：

16、在本申请实施方式中，在得到视频中每一帧的天空分割结果后，根据天空分割的历史信息确定每一帧的误识别情况，继而对误识别的帧进行相应的调整处理。继而在不改变模型参数的前提下，处理误分割现象，解决了模型输出在视频帧间渲染异常、抖动的问题，提升视频美观和观看者的观看体验。

技术特征：

1.一种视频天空区域调色方法，其特征在于，所述方法包括：

2.根据权利要求1所述的方法，其特征在于，所述根据所述初始二值图对n-1张视频帧图像进行n-1次天空区域识别处理，得到n-1张最终二值图，包括：

3.根据权利要求2所述的方法，其特征在于，所述方法还包括：

4.根据权利要求1所述的方法，其特征在于，在所述对所述n张视频帧图像中的第一张视频帧图像进行天空区域识别，得到初始二值图之后，所述方法还包括：

5.根据权利要求4所述的方法，其特征在于，所述松弛变量通过以下公式确定：

6.根据权利要求4或5所述的方法，其特征在于，所述第一阈值通过以下公式确定：

7.一种视频天空区域调色装置，其特征在于，所述装置包括：

8.根据权利要求7所述的装置，其特征在于，在所述根据所述初始二值图对n-1张视频帧图像进行n-1次天空区域识别处理，得到n-1张最终二值图方面，所述处理模块，具体用于：

9.一种电子设备，其特征在于，包括处理器、存储器、通信接口以及一个或多个程序，其中，所述一个或多个程序被存储在所述存储器中，并且被配置由所述处理器执行，所述一个或多个程序包括用于执行权利要求1-6任一项所述的方法中的步骤的指令。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行以实现如权利要求1-6任一项所述的方法。

技术总结
本申请公开了一种视频天空区域调色方法、装置、电子设备及存储介质，其中，方法包括：对视频数据进行分帧处理，得到N张视频帧图像；对所述N张视频帧图像中的第一张视频帧图像进行天空区域识别，得到初始二值图；根据所述第一张视频帧图像的尺寸信息，确定第一阈值；根据所述初始二值图和所述第一阈值，对N‑1张视频帧图像进行N‑1次天空区域识别处理，得到N‑1张最终二值图，其中，所述N‑1张视频帧图像为所述N张视频帧图像中除去所述第一张视频帧图像后剩下的N‑1张图像；根据所述初始二值图和所述N‑1张最终二值图，对所述N张视频帧图像进行分天空区域调色，得到调色后的视频数据。

技术研发人员：王浩帆
受保护的技术使用者：书行科技（北京）有限公司
技术研发日：
技术公布日：2024/3/31