[发明专利]去马赛克方法、装置、设备及存储介质

权利要求书

1.一种去马赛克方法，其特征在于，所述方法包括下述步骤：

获取拜耳格式图像；

确定所述拜耳格式图像G通道上每个缺失像素的插值方向是否明显；

采用预设的插值方式，对插值方向明显的G通道缺失像素进行插值；

按照预设的插值方向选取规则，为插值方向不明显的G通道缺失像素选取插值方向，根据选取的插值方向进行插值，得到完整的G通道信息；

基于所述完整的G通道信息对R通道和B通道进行插值，得到RGB图像并输出；

确定所述拜耳格式图像G通道上每个缺失像素的插值方向是否明显的步骤，包括：

基于G通道当前缺失像素为中心的像素阵列进行G通道当前缺失像素的水平方向和垂直方向的梯度计算，并基于G通道当前缺失像素为中心的像素阵列进行G通道当前缺失像素的水平方向和垂直方向的插值估算；

判断G通道当前缺失像素是否满足预设条件，所述预设条件为所述水平方向和垂直方向的梯度计算值的接近度小于预设的第二阈值、且所述水平方向和垂直方向的插值估算值的差异度大于预设的第三阈值；

若满足，则判定G通道当前缺失像素为插值方向不明显的缺失像素，否则判定G通道当前缺失像素为插值方向明显的缺失像素；

所述基于G通道当前缺失像素为中心的像素阵列进行G通道当前缺失像素的水平方向和垂直方向的插值估算包括：

采用插值阶次自适应的方式基于G通道当前缺失像素为中心的像素阵列进行G通道当前缺失像素的第一预设方向的插值估算，在所述第一预设方向的任一方向上，若G通道当前缺失像素在C通道的梯度和同方向上的G通道的梯度的相似度小于预设的第一阈值，则采用一阶插值法进行插值估算，否则采用零阶插值法进行插值估算，所述第一预设方向包括东、南、西、北四个方向，当所述G通道当前缺失像素为R像素时，所述C通道为R通道，当所述G通道当前缺失像素为B像素时，所述C通道为B通道；

对东、西两个方向的插值估算值进行加权平均，得到水平方向的插值估算值，对南、北两个方向的插值估算值进行加权平均，得到垂直方向的插值估算值；

采用预设的插值方式，对插值方向明显的G通道缺失像素进行插值的步骤，包括：

若所述水平方向的梯度计算值与所述第一阈值之和小于所述垂直方向的梯度计算值，则采用所述水平方向的插值估算值对G通道当前缺失像素进行插值；

若所述垂直方向的梯度计算值与所述第一阈值之和小于所述水平方向的梯度计算值，则采用所述垂直方向的插值估算值对G通道当前缺失像素进行插值；

否则，采用东、南、西、北四个方向的插值估算值的加权平均值对G通道当前缺失像素进行插值。

2.如权利要求1所述的方法，其特征在于，每个方向的插值权重的计算公式是随该方向的梯度计算值的增大而减小的递减函数，所述插值权重的计算公式如下：

其中，E、S、W、N分别表示东、南、西、北四个方向，w_d(i,j)表示G通道当前缺失像素沿东、南、西或北向的插值权重，GM_d(i,j)表示G通道当前缺失像素沿东、南、西或北向的梯度计算值，i和j分别表示G通道当前缺失像素的行号和列号。

3.如权利要求1所述的方法，其特征在于，按照预设的插值方向选取规则，为插值方向不明显的G通道缺失像素选取插值方向的步骤，包括：

按照色差通道的能量大小，为插值方向不明显的G通道缺失像素选取插值方向；

按照色差通道的能量大小，为插值方向不明显的G通道缺失像素选取插值方向的步骤，包括：

将能量最小的色差通道对应的插值方向设置为G通道当前缺失像素的插值方向，其中，所述色差通道的能量大小基于色差通道的方差、熵或梯度获取。

4.如权利要求1所述的方法，其特征在于，基于所述完整的G通道信息对R通道和B通道进行插值的步骤，包括：

对C通道的每个有效像素进行色差提取，得到降采样的色差通道信息；

以色差通道当前缺失像素为中心取第一邻域窗，根据所述第一邻域窗内的中心像素与有效像素的关联性和相似性对所述降采样的色差通道信息进行双边滤波，以恢复完整的色差通道信息，其中，所述第一邻域窗内的中心像素与有效像素的关联性根据完整的G通道信息获得；

基于所述完整的G通道信息和所述完整的色差通道信息对R通道和B通道进行插值。