[发明专利]图像增强方法、装置、电子设备及存储介质

说明书

本申请实施例公开了一种图像增强方法、装置、设备及存储介质，属于图像处理领域。所述方法包括：获取待处理的目标图像；对所述目标图像进行语义分割，得到所述目标图像的n个语义区域，n为大于1的整数；根据参考图像对所述目标图像的n个语义区域按照各自对应的影响因子进行风格转移，得到增强后的目标图像；其中，所述影响因子用于表示对所述语义区域进行增强时的程度。本申请实施例能够实现对目标图像中的各个语义区域分别进行不同程度的风格转移，从而解决了对目标图像进行整体性的风格转移时增强效果较差的问题；达到了对不同的语义区域按照合适的程度进行增强，使得整张目标图像得到较好的增强性效果。

技术领域

本申请实施例涉及机器学习领域，特别涉及一种图像增强方法、装置、电子设备及存储介质。

背景技术

在夜晚拍摄照片时，需要拍摄者具有许多摄影技巧以及高质量的相机。现在大多数的智能手机，由于设备本身的体积、硬件参数和操作方式存在局限性，很难拍摄出专业的夜景照片。

相关技术中提供了基于风格迁移的图像增强方法，该方法包括：在图像增强程序中输入一张夜景图像，并且由用户在图像增强程序中手动选择一张风格模板图像，图像增强程序会基于深度学习的方法，计算输入的夜景图像和风格模板图像之间的风格损失以及内容损失，经过若干个迭代运算后输出夜景图像。输出的夜景图像在视觉上按照风格模板图像进行了一个整体性的风格化迁移，从而得到了增强。

由于夜景图像在图像内容上有很大的变化，可能会有非常复杂的结构。例如许多建筑物在白天看起来很相似，但是在晚上，它们可以有完全不相同的窗户和壁灯。更不用提许多其它不同类型的物体和光线，比如道路、天空、汽车等等。相关技术中的图像增强方法并不能适应不同内容的夜景图像增强。

发明内容

本申请实施例提供了一种图像增强方法、装置、设备及存储介质，可以解决相关技术中的图像增强方法不能适应不同内容的夜景图像增强，在某些场景下增强效果较差的问题。所述技术方案如下：

根据本申请的一个方面，提供了一种图像增强方法，所述方法包括：

获取待处理的目标图像；

对所述目标图像进行语义分割，得到所述目标图像的n个语义区域，n为大于1的整数；

根据参考图像对所述目标图像的n个语义区域按照各自对应的影响因子进行风格转移，得到增强后的目标图像；

其中，所述影响因子用于表示对所述语义区域进行增强时的程度。

根据本申请的另一方面，提供了一种图像增强装置，所述装置包括：

图像获取模块，用于获取待处理的目标图像；

语义分割模块，用于对所述目标图像进行语义分割，得到所述目标图像的n个语义区域，n为大于1的整数；

风格转移模块，用于根据参考图像对所述目标图像的n个语义区域按照各自对应的影响因子进行风格转移，得到增强后的目标图像；

其中，所述影响因子用于表示对所述语义区域进行增强时的程度。

根据本申请的另一方面，提供了一种电子设备，所述电子设备包括存储器和处理器；所述存储器存储有至少一条指令，所述至少一条指令由所述处理器加载并执行以实现如上所述的图像增强方法。

根据本申请的另一方面，提供了一种计算机可读存储介质，所述存储介质中存储有至少一条指令，所述至少一条指令由处理器加载并执行以实现如上所述的图像增强方法。

本申请实施例提供的技术方案带来的有益效果至少包括：