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

一种利用红外图去除屏下RGB图像彩虹状眩光方法，包括如下步骤：步骤S1：获取第一RGB图像和对应的第一红外图，将所述第一红外图与所述第一RGB图像对齐得到第二红外图；步骤S2：基于所述第一RGB图像和所述第二红外图，利用深度神经网络模型去除所述第一RGB图像中的所述至少两个相连的光源及其彩虹状眩光，得到第二RGB图像；步骤S3：将所述第一RGB图像中的所述至少两个相连的光源不均匀地与所述第二RGB图像融合生成第三RGB图像；步骤S4：利用神经网络模型消除所述第三RGB图像中的阶梯式缺陷。本发明专利技术针对多光源情况，既能有效去除彩虹状眩光，又能恢复因去除光源而丢失的信息，保证去除光源后的图像更加真实全面。除光源后的图像更加真实全面。除光源后的图像更加真实全面。

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


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

技术介绍

[0002]近年来，随着手机全面屏理念的快速发展，不少手机厂商都在尝试将前置摄像头放在显示屏后面，以实现真正全面屏的效果。但是，将前置摄像头放在显示屏后面会带来新的问题，即屏下摄像头成像质量严重下降。特别地，当视野内出现较强光源时，屏下摄像头拍摄的RGB彩色图像会出现严重的彩虹状眩光，这种眩光一圈一圈围绕在光源周边，遮盖了背景信息，带给人的体验很不好。
[0003]目前人们开始针对屏下摄像头带来的质量问题进行研究，但彩虹状眩光问题目前还没有针对性的解决方案。之前也有人研究如何去除RGB图像眩光，但针对的均是非屏下摄像头拍摄的图像，其眩光模式和屏下摄像头拍摄的图像中彩虹状眩光区别很大。
[0004]传统的眩光由于强光在传感器内光学元件的作用下经过散射和反射形成。而屏下摄像头中的眩光不仅由传感器内的光学元件的特性造成，还由于显示屏的影响而造成光本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种利用红外图去除屏下RGB图像彩虹状眩光方法，其特征在于，包括如下步骤：步骤S1：获取第一RGB图像和对应的第一红外图，将所述第一红外图与所述第一RGB图像对齐得到第二红外图；其中，所述第一RGB图像至少包含至少两个相连的光源及其彩虹状眩光，所述第一红外图至少包含两个相连的光源；步骤S2：基于所述第一RGB图像和所述第二红外图，利用深度神经网络模型去除所述第一RGB图像中的所述至少两个相连的光源及其彩虹状眩光，得到第二RGB图像；步骤S3：将所述第一RGB图像中的所述至少两个相连的光源不均匀地与所述第二RGB图像融合生成第三RGB图像；步骤S4：利用神经网络模型消除所述第三RGB图像中的阶梯式缺陷。2.根据权利要求1所述的一种利用红外图去除屏下RGB图像彩虹状眩光方法，其特征在于，在步骤S1中，对所述第二红外图周边进行padding以使得所述第二红外图的分辨率和所述第一RGB图一致。3.根据权利要求1所述的一种利用红外图去除屏下RGB图像彩虹状眩光方法，其特征在于，所述深度神经网络模型分别提取所述第一RGB图像的信息和所述第二红外图的信息，经过特征融合转换之后得到没有光源和眩光的所述第二RGB图像，以有效去除眩光和恢复所述所述第二RGB图像中丢失的信息。4.根据权利要求1所述的一种利用红外图去除屏下RGB图像彩虹状眩光方法，其特征在于，所述深度神经网络模型在训练时，通过数据增广拟合出多个光源。5.根据权利要求1所述的一种利用红外图去除屏下RGB图像彩虹状眩光方法，其特征在于，所述步骤S3包括：步骤S31：在所述第一RGB图像中将光源分割，生成掩膜；步骤S32：对所述掩膜进行不均匀膨胀，每膨胀一圈就设置一个权重，则从光源边缘扩散到有效边缘的过程中，权重从1逐渐衰减到0，...

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