去马赛克的方法、计算机存储介质和电子设备与流程

本发明涉及图像处理，尤其是涉及一种去马赛克的方法、计算机存储介质和电子设备。

背景技术：

1、目前主流的光学成像传感器会使用彩色滤波阵列对输入光线进行采样，最常用的是拜耳滤色镜，其采样的拜耳图像中rgb按照规律交替排列，每个像素点只有一个颜色值，其中，g像素的数量为r像素或b像素的两倍。相关技术中，采用去马赛克算法对拜耳图像中各个像素点重建rgb三原色，但是，因传感器采集到的数据受制于滤波器的采样频率，无法准确重建拜耳图像中的高频信息，存在较多的伪彩和不连续点。

技术实现思路

1、本发明旨在至少解决现有技术中存在的技术问题之一。为此，本发明的一个目的在于提出一种去马赛克的方法，采用该方法可以改善去除马赛克后的彩色图像中的伪彩问题，提高图像质量。

2、本发明的目的之二在于提出一种计算机存储介质。

3、本发明的目的之三在于提出一种电子设备。

4、为了解决上述问题，本发明第一方面实施例提供一种去马赛克的方法，包括：获取拜耳格式图像，其中，所述拜耳格式图像包括g像素点和非g像素点，所述非g像素点包括r像素点和b像素点；对所述拜耳格式图像中非g像素点进行g分量插值，以获得水平方向的第一非g像素点插值图像和第一g分量插值图像，以及获得垂直方向的第二非g像素点插值图像和第二g分量插值图像；根据所述第一非g像素点插值图像和所述第一g分量插值图像通过导向滤波法获得水平方向的第一g分量滤波图像，以及，根据所述第二非g像素点插值图像和所述第二g分量插值图像通过导向滤波法获得垂直方向的第二g分量滤波图像；根据所述第一g分量滤波图像和所述拜耳格式图像获得第一g分量残差值，以及根据所述第二g分量滤波图像和所述拜耳格式图像获得第二g分量残差值；根据第一g分量残差值、第二g分量残差值和所述拜耳格式图像中的g像素点图像获得g通道图像；重建r通道图像和b通道图像；根据所述g通道图像、所述r通道图像和所述b通道图像获得去除马赛克后的彩色图像。

5、根据本发明实施例的去马赛克的方法，通过在对拜耳格式图像中非g像素点进行g分量插值后，引入导向滤波法来获取水平方向的第一g分量滤波图像和垂直方向的第二g分量滤波图像，由此能够很好的恢复拜耳图像中的颜色信息，有效减少图像中不自然颜色的出现，进而在此基础上再进行颜色残差计算，以第一g分量残差值、第二g分量残差值和拜耳格式图像中的g像素点图像来获得g通道图像，有效改善去除马赛克后的彩色图像中的伪彩问题，提高彩色图像质量。

6、在一些实施例中，所述g像素点包括在水平方向上与所述r像素点相邻的gr像素点和在水平方向上与所述b像素点相邻的gb像素点，所述第一非g像素点插值图像包括第一r像素点插值图像和第一b像素点插值图像，所述第一g分量插值图像包括第一gr分量插值图像和第一gb分量插值图像，所述第二非g像素点插值图像包括第二r像素点插值图像和第二b像素点插值图像，所述第二g分量插值图像包括第二gr分量插值图像和第二gb分量插值图像，

7、对所述拜耳格式图像中非g像素点进行g分量插值，以获得水平方向的第一非g像素点插值图像和第一g分量插值图像，以及获得垂直方向的第二非g像素点插值图像和第二g分量插值图像，包括：

8、对所述拜耳格式图像中的r像素点图像和gr像素点图像在水平方向上分别进行卷积处理，以获得r像素点的第一gr分量卷积图像和gr像素点的第一r分量卷积图像；

9、将所述第一gr分量卷积图像与所述r像素点图像进行组合，以获得所述第一gr分量插值图像，以及将所述第一r分量卷积图像与所述gr像素点图像进行组合，以获得所述第一r像素点插值图像；

10、对所述拜耳格式图像中的b像素点图像和gb像素点图像在水平方向上分别进行卷积处理，以获得b像素点的第一gb分量卷积图像和gb像素点的第一b分量卷积图像；

11、将所述第一gb分量卷积图像与所述b像素点图像进行组合，以获得所述第一gb分量插值图像，以及将所述第一b分量卷积图像与所述gb像素点图像进行组合，以获得所述第一b像素点插值图像；

12、对所述拜耳格式图像中的r像素点图像和gb像素点图像在垂直方向上分别进行卷积处理，以获得r像素点的第二gb分量卷积图像和gb像素点的第二r分量卷积图像；

13、将所述第二gb分量卷积图像与所述r像素点图像进行组合，以获得所述第二gb分量插值图像，以及将所述第二r分量卷积图像与所述gb像素点图像进行组合，以获得所述第二r像素点插值图像；

14、对所述拜耳格式图像中的b像素点图像和gr像素点图像在垂直方向上分别进行卷积处理，以获得b像素点的第二gr分量卷积图像和gr像素点的第二b分量卷积图像；

15、将所述第二gr分量卷积图像与所述b像素点图像进行组合，以获得所述第二gr分量插值图像，以及将所述第二b分量卷积图像与所述gr像素点图像进行组合，以获得所述第二b像素点插值图像。

16、在一些实施例中，所述第一g分量滤波图像包括第一gr分量滤波图像和第一gb分量滤波图像，根据所述第一非g像素点插值图像和所述第一g分量插值图像通过导向滤波法获得水平方向的第一g分量滤波图像，包括：将所述第一r像素点插值图像作为导向图，通过所述导向滤波法对所述第一gr分量插值图像进行滤波处理，以获得所述第一gr分量滤波图像；将所述第一b像素点插值图像作为导向图，通过所述导向滤波法对所述第一gb分量插值图像进行滤波处理，以获得所述第一gb分量滤波图像。

17、在一些实施例中，所述第二g分量滤波图像包括第二gr分量滤波图像和第二gb分量滤波图像，根据所述第二非g像素点插值图像和所述第二g分量插值图像通过导向滤波法获得垂直方向的第二g分量滤波图像，包括：将所述第二r像素点插值图像作为导向图，通过所述导向滤波法对所述第二gb分量插值图像进行滤波处理，以获得所述第二gb分量滤波图像；将所述第二b像素点插值图像作为导向图，通过所述导向滤波法对所述第二gr分量插值图像进行滤波处理，以获得所述第二gr分量滤波图像。

18、在一些实施例中，所述第一g分量残差值包括第一gr分量残差值和第一gb分量残差值，所述第二g分量残差值包括第二gr分量残差值和第二gb分量残差值，根据所述第一g分量滤波图像和所述拜耳格式图像获得第一g分量残差值，以及根据所述第二g分量滤波图像和所述拜耳格式图像获得第二g分量残差值，包括：将所述第一gr分量滤波图像与所述拜耳格式图像作差值处理，以获得所述第一gr分量残差值；将所述第一gb分量滤波图像与所述拜耳格式图像作差值处理，以获得所述第一gb分量残差值；将所述第二gr分量滤波图像与所述拜耳格式图像作差值处理，以获得所述第二gr分量残差值；将所述第二gb分量滤波图像与所述拜耳格式图像作差值处理，以获得所述第二gb分量残差值。

19、在一些实施例中，根据第一g分量残差值、第二g分量残差值和所述拜耳格式图像中的g像素点图像获得g通道图像，包括：对所述第一g分量残差值和所述第二g分量残差值分别进行低通滤波处理；根据低通滤波处理后的第一g分量残差值和所述第一g分量插值图像获得第一g分量修正图像；根据低通滤波处理后的第二g分量残差值和所述第二g分量插值图像获得第二g分量修正图像；根据所述拜耳格式图像中的g像素点图像、所述第一g分量修正图像和所述第二g分量修正图像获得所述g通道图像。

20、在一些实施例中，第一g分量修正图像包括第一gr分量修正图像和第一gb分量修正图像，第二g分量修正图像包括第二gr分量修正图像和第二gb分量修正图像，根据所述拜耳格式图像中的g像素点图像、所述第一g分量修正图像和所述第二g分量修正图像获得所述g通道图像，包括：将所述g像素点图像、所述第一gr分量修正图像和所述第一gb分量修正图像进行组合，以获得水平方向下的第一g分量图像；将所述g像素点图像、所述第二gr分量修正图像和所述第二gb分量修正图像进行组合，以获得垂直方向下的第二g分量图像；计算所述拜耳格式图像中每个像素点在水平方向下的第一梯度值和在垂直方向下的第二梯度值；根据所述第一梯度值和所述第二梯度值对所述第一g分量图像和所述第二g分量图像进行加权平均，以获得所述g通道图像。

21、在一些实施例中，重建r通道图像和b通道图像，包括：计算所述拜耳格式图像中每个像素点的r分量在插值方向下的第三梯度值和每个像素点的b分量在插值方向下的第四梯度值，其中，所述插值方向包括水平方向、垂直方向、45度方向以及135度方向；根据所述第三梯度值对r分量进行插值，以重建所述r通道图像；根据所述第四梯度值对b分量进行插值，以重建所述b通道图像。

22、本发明第二方面实施例提供一种计算机存储介质，其上存储有计算机程序，其中，所述计算机程序被处理器执行时实现上述实施例所述的去马赛克的方法。

23、本发明第三方面实施例提供一种电子设备，包括：至少一个处理器；与至少一个所述处理器通信连接的存储器；其中，所述存储器中存储有可被至少一个所述处理器执行的计算机程序，至少一个所述处理器执行所述计算机程序时实现上述实施例所述的去马赛克的方法。

24、根据本发明实施例的电子设备，通过处理器执行上述实施例提供的去马赛克的方法，可以改善去除马赛克后的彩色图像中的伪彩问题，提高图像质量。

25、本发明的附加方面和优点将在下面的描述中部分给出，部分将从下面的描述中变得明显，或通过本发明的实践了解到。