一种定向去除屏下RGB图像彩虹状眩光方法、设备及存储介质技术

一种定向去除屏下RGB图像彩虹状眩光方法，其特征在于，包括如下步骤：步骤S1：获取第一RGB图像，利用深度神经网络模型不定向去除彩虹状眩光，得到第二RGB图像；其中，所述不定向是指同时去除眩光和光源；步骤S2：将所述第一RGB图像中的光源及周边信息不均匀地与所述第二RGB图像融合生成第三RGB图像；步骤S3：利用神经网络模型消除所述第三RGB图像中的阶梯式缺陷。本发明专利技术先利用深度神经网络模型不定向去除彩虹状眩光，再将处理前后的图像进行融合得到带有光源的第三RGB图像，再消除第三RGB图像上的阶梯式缺陷，得到最终的RGB图像，从而有效去除屏下相机拍摄的RGB图像中的眩光。效去除屏下相机拍摄的RGB图像中的眩光。效去除屏下相机拍摄的RGB图像中的眩光。

【技术实现步骤摘要】
一种定向去除屏下RGB图像彩虹状眩光方法、设备及存储介质


[0001]本专利技术涉及图像处理领域，具体地，涉及一种定向去除屏下RGB图像彩虹状眩光方法、设备及存储介质。

技术介绍

[0002]近年来，随着手机全面屏理念的快速发展，不少手机厂商都在尝试将前置摄像头放在显示屏后面，以实现真正全面屏的效果。但是，将前置摄像头放在显示屏后面会带来新的问题，即屏下摄像头成像质量严重下降。特别地，当视野内出现较强光源时，屏下摄像头拍摄的RGB彩色图像会出现严重的彩虹状眩光，这种眩光一圈一圈围绕在光源周边，遮盖了背景信息，带给人的体验很不好。
[0003]目前人们开始针对屏下摄像头带来的质量问题进行研究，但彩虹状眩光问题目前还没有针对性的解决方案。之前也有人研究如何去除RGB图像眩光，但针对的均是非屏下摄像头拍摄的图像，其眩光模式和屏下摄像头拍摄的图像中彩虹状眩光区别很大。
[0004]该问题的难点在于，如何在去除光源周边彩虹状眩光的同时保留光源，同时还要保证光源边缘和周围背景自然过渡，使得图像在定向去除眩光之后整体上能保证自然的效果。...

【技术保护点】

【技术特征摘要】
1.一种定向去除屏下RGB图像彩虹状眩光方法，其特征在于，包括如下步骤：步骤S1：获取第一RGB图像，利用深度神经网络模型不定向去除彩虹状眩光，得到第二RGB图像；其中，所述不定向是指同时去除眩光和光源；步骤S2：将所述第一RGB图像中的光源及周边信息不均匀地与所述第二RGB图像融合生成第三RGB图像；步骤S3：利用神经网络模型消除所述第三RGB图像中的阶梯式缺陷。2.根据权利要求1所述的一种定向去除屏下RGB图像彩虹状眩光方法，其特征在于，在步骤S1中，所述深度神经网络模型由一个光源打开/关闭的一一对应的数据集训练获得。3.根据权利要求1所述的一种定向去除屏下RGB图像彩虹状眩光方法，其特征在于，所述深度神经网络模型采用Encoder
‑
BottleneckBlock
‑
Decoder架构，其中，Encoder部分利用预训练的VGGNet提取图像特征，BottleneckBlock部分采用多个残差模块叠加，且所述残差模块均带通道注意力机制，Decoder部分将所述BottleneckBlock部分的输出上采样到原图分辨率。4.根据权利要求1所述的一种定向去除屏下RGB图像彩虹状眩光方法，其特征在于，所述步骤S2包括：步骤S21：在所述第一RGB图像中将光源分割，生成掩膜；步骤S22：对所述掩膜进行膨胀，每膨胀一圈就设置一个权重，则从光源边缘向外扩散的过程中，权重从1.0逐渐衰减到0.0，从而得到一张单通道...

【专利技术属性】
技术研发人员：曾曌，邵小飞，朱程，李向南，汪博，朱力，吕方璐，
申请(专利权)人：深圳市光鉴科技有限公司，
类型：发明
国别省市：