【技术实现步骤摘要】
图像增强模型的训练、图像增强方法及电子设备
[0001]本专利技术涉及图像处理
,具体涉及图像增强模型的训练、图像增强方法及电子设备。
技术介绍
[0002]低照度下的高质量成像是一项具有挑战性但意义重大的任务。一方面,它在许多领域中发挥着至关重要的作用,例如监控、汽车驾驶、智能手机摄影等。另一方面,由于低照度环境下的曝光和照明,从而导致极低的信噪比(SNR)和相应的成像质量严重下降。针对该问题,最常见的一类方法是使用补光灯,但会导致光污染及行人和司机的眩晕感,而且低照度下不合理的灯光会给人体造成不适感。
技术实现思路
[0003]有鉴于此,本专利技术实施例提供了一种图像增强模型的训练、图像增强方法及电子设备,以解决低照度环境下的成像问题。
[0004]根据第一方面,本专利技术实施例提供了一种图像增强方法,包括:
[0005]获取样本图像组,所述样本图像组包括双目相机采集的样本图像以及基准图像,所述样本图像为所述双目相机在第一照度下进行近红外补光后采集的图像,所述样本图像包括RGB图像和 ...
【技术保护点】
【技术特征摘要】
1.一种图像增强模型的训练方法,其特征在于,包括:获取样本图像组,所述样本图像组包括双目相机采集的样本图像以及基准图像,所述样本图像为所述双目相机在第一照度下进行近红外补光后采集的图像,所述样本图像包括RGB图像和NIR图像,所述RGB图像和所述NIR图像位于所述双目相机的不同视场,所述基准图像为所述双目相机在第二照度下采集的RGB图像,所述第一照度低于第二照度;将所述RGB图像以及所述NIR图像输入图像增强模型中,以利用所述图像增强模型对所述RGB图像以及所述NIR图像进行视场对齐以及增强重建处理,得到图像增强预测图像,其中,所述视场对齐是将所述RGB图像和所述NIR图像在亮度通道上的特征相似关系作用于所述RGB图像和/或所述NIR图像实现的;基于所述图像增强预测图像与所述基准图像进行损失计算,以对所述图像增强模型的参数进行更新,确定目标图像增强模型。2.根据权利要求1所述的方法,其特征在于,所述将所述RGB图像以及所述NIR图像输入图像增强模型中,以利用所述图像增强模型对所述RGB图像以及所述NIR图像进行视场对齐,包括:利用所述图像增强模型中的两个特征提取单元分别对所述RGB图像以及所述NIR图像进行亮度通道上的特征提取得到第一参考特征和第二参考特征;利用所述图像增强模型中的两个语义特征提取单元分别对所述RGB图像以及所述NIR图像进行语义特征提取得到第一语义特征和第二语义特征,所述特征提取单元与所述语义特征提取单元的结构相同且参数不同;利用所述图像增强模型计算所述第一参考特征和第二参考特征间的特征相似关系,并基于所述特征相似关系对所述第一语义特征和/或所述第二语义特征进行预设视场下的特征对齐得到对齐特征以实现所述视场对齐。3.根据权利要求2所述的方法,其特征在于,所述计算所述第一参考特征和第二参考特征间的特征相似关系,并基于所述特征相似关系对所述第一语义特征和/或所述第二语义特征进行预设视场下的特征对齐得到对齐特征以实现所述视场对齐,包括:计算所述第一参考特征和所述第二参考特征间的相似度,得到相似度矩阵;将所述相似度矩阵内元素的值映射为预设范围内的值,确定所述特征相似关系。4.根据权利要求2所述的方法,其特征在于,所述计算所述第一参考特征和第二参考特征间的特征相似关系,包括:获取所述RGB图像的第一参考特征中第一像素点所在的目标行;从所述NIR图像的第二参考特征中提取与所述目标行对应的行的第二像素点;计算所述第一像素点与所述第二像素点的之间的相似度,确定所述特征相似关系。5.根据权利要求2所述的方法,其特征在于,所述对齐特征包括所述第一语义特征对应的第一对齐特征以及所述第二语义特征对应的第二对齐特征,...
【专利技术属性】
技术研发人员:张艳丽,李瑮,毛晓蛟,
申请(专利权)人:苏州科达科技股份有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。