【技术实现步骤摘要】
一种图像增强方法、装置及电子设备
[0001]本申请涉及计算机视觉图像技术,尤其涉及一种图像增强方法、装置及电子设备。
技术介绍
[0002]近年来,随着显示设备的显示亮度和显示色域的不断提高,对高对比度,广色域的图像的要求也越来越高。受限于拍摄场景与拍摄设备等原因,在很多情况下光源的位置和光照条件是不可控的,这种图像上曝光不足或曝光过度的区域几乎看不到细节,目前很多拍摄或录制所得到的图像可能会存在过曝、欠曝、饱和度低、对比度低等问题,造成不好的视觉体验。
[0003]为了在不使用更高端的设备重新拍摄制作视频资源的前提下,使人们在具有高显示能力的设备上体验高质量画面,相关技术中通常采用图像增强方法对图像进行处理增强对比度与颜色。然而图像增强方法会对所有输入的图像进行增强。然而,非自然图像一般由作者按照自身意图进行制作,不受拍摄设备、条件等的限制,因此该类图像对比度与颜色已经比较合适。如果对非自然图像不加以选择地进行增强,则会导致色彩失真、对比度失真、细节丢失等问题。
技术实现思路
[0004]本申请实 ...
【技术保护点】
【技术特征摘要】
1.一种图像增强方法,其特征在于,所述方法包括:获取待增强图像,将所述待增强图像输入预先训练完成的非自然图像识别模型中,获取所述非自然图像识别模型输出的所述待增强图像为非自然图像及为自然图像的概率;并获取所述待增强图像对应的增强图像;针对所述增强图像中的每个像素点,获取所述增强图像中该像素点的第一像素值及所述待增强图像中该像素点对应的像素点的第二像素值,将所述待增强图像为非自然图像及为自然图像的概率分别作为所述第二像素值及所述第一像素值对应的权重,确定该像素点在融合后的目标图像中对应像素点的目标像素值。2.根据权利要求1所述的方法,其特征在于,所述获取所述待增强图像对应的增强图像包括:将所述待增强图像输入内容增强模型,获取所述内容增强模型输出的增强对比度及颜色的待处理图像;将所述待处理图像输入感知增强模型,获取所述感知增强模型输出的增强图像;其中,所述增强图像为增强采集待增强图像的场景相关的感知信息的图像。3.根据权利要求1所述的方法,其特征在于,所述获取待增强图像之后,所述将所述待增强图像输入预先训练完成的非自然图像识别模型中之前,所述方法还包括:将所述待增强图像输入特征提取网络中,获取所述特征提取网络输出的所述待增强图像的图像全局特征;所述将所述待增强图像输入预先训练完成的非自然图像识别模型中包括:将所述待增强图像的图像全局特征输入预先训练完成的非自然图像识别模型中。4.根据权利要求1所述的方法,其特征在于,所述非自然图像识别模型通过以下方式训练:获取第一样本集中任一第一样本图像及第一样本图像为非自然图像及为自然图像的标注概率;将所述第一样本图像输入原始识别模型,获取所述原始识别模型输出的所述第一样本图像为非自然图像及为自然图像的输出概率;根据所述标注概率、所述输出概率及第一损失函数,确定所述第一样本图像对应的第一损失值;根据所述第一损失值,对所述原始识别模型进行训练。5.根据权利要求4所述的方法,其特征在于,所述第一损失函数为:l
c
=
‑
[y
i
·
log(p
i
)+(1
‑
y
i
)
·
log(1
‑
p
i
)]其中,所述y
i
为第一样本图像i为非自然图像的标注概...
【专利技术属性】
技术研发人员:顾博文,查林,姜建德,雒旭鹏,路文,高新波,
申请(专利权)人:青岛信芯微电子科技股份有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。