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

说明书

本申请公开了一种图像处理方法、装置、电子设备及存储介质，该图像处理方法包括：获取待处理图像，所述待处理图像包括阴影区域；将所述待处理图像输入至预先训练的阴影抠图模型，获得所述阴影抠图模型输出的阴影抠图，所述阴影抠图用于为所述待处理图像对应的阴影比例因子所表示的图像，所述阴影抠图模型为基于阴影样本图像以及无阴影样本图像，对循环一致性生成对抗网络训练得到；将所述待处理图像与所述阴影抠图进行像素级相除，获得去阴影后的图像。本方法在实现对图像中的阴影进行去除时，可以达到较佳的阴影去除效果。

技术领域

本申请涉及图像处理技术领域，更具体地，涉及一种图像处理方法、装置、电子设备及存储介质。

背景技术

随着科技水平和生活水平的快速进步，电子设备被人们广泛地使用。通常电子设备具有图像采集功能，可以方便人们采集日常生活和工作中所需的图像。然而在图像检测、物体识别、图像修复、分割抠图、文档矫正等领域进行图像采集时，由于自然光照和非自然光照的影响，会使得采集到的图像往往都会含有阴影，阴影会对后续的处理方法造成严重的影响。

发明内容

鉴于上述问题，本申请提出了一种图像处理方法、装置、电子设备及存储介质。

第一方面，本申请实施例提供了一种像处理方法，所述方法包括：获取待处理图像，所述待处理图像包括阴影区域；将所述待处理图像输入至预先训练的阴影抠图模型，获得所述阴影抠图模型输出的阴影抠图，所述阴影抠图为所述待处理图像对应的阴影比例因子所表示的图像，所述阴影抠图模型为基于阴影样本图像以及无阴影样本图像，对循环一致性生成对抗网络训练得到；将所述待处理图像与所述阴影抠图进行像素级相除，获得去阴影后的图像。

第二方面，本申请实施例提供了一种图像处理装置，所述装置包括：第一图像获取模块、第二图像获取模块以及第三图像获取模块，其中，所述第一图像获取模块用于获取待处理图像，所述待处理图像包括阴影区域；所述第二图像获取模块用于将所述待处理图像输入至预先训练的阴影抠图模型，获得所述阴影抠图模型输出的阴影抠图，所述阴影抠图为所述待处理图像对应的阴影比例因子所表示的图像，所述阴影抠图模型为基于阴影样本图像以及无阴影样本图像，对循环一致性生成对抗网络训练得到；所述第三图像获取模块用于将所述待处理图像与所述阴影抠图进行像素级相除，获得去阴影后的图像。

第三方面，本申请实施例提供了一种电子设备，包括：一个或多个处理器；存储器；一个或多个应用程序，其中所述一个或多个应用程序被存储在所述存储器中并被配置为由所述一个或多个处理器执行，所述一个或多个程序配置用于执行上述第一方面提供的图像处理方法。

第四方面，本申请实施例提供了一种计算机可读取存储介质，所述计算机可读取存储介质中存储有程序代码，所述程序代码可被处理器调用执行上述第一方面提供的图像处理方法。

本申请提供的方案，通过获取包括阴影区域的待处理图像，将待处理图像输入至预先训练的阴影抠图模型，获得阴影抠图模型输出的阴影抠图，该阴影抠图为待处理图像对应的阴影比例因子所表示的图像，并且该阴影抠图模型是基于阴影样本图像以及无阴影样本图像，对循环一致性生成对抗网络训练得到，再将待处理图像与阴影抠图进行像素级相除，获得去阴影后的图像，由于阴影抠图模型是对循环一致性生成网络训练得到，可以提升阴影抠图模型获取阴影抠图的准确性，进而在利用阴影抠图对待处理图像进行阴影去除时，能够提升阴影去除的效果。

附图说明

为了更清楚地说明本申请实施例中的技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本申请的一些实施例，对于本领域技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。

图1示出了根据本申请一个实施例的图像处理方法流程图。

图2示出了本申请实施例提供的图像处理方法的原理示意图。

图3示出了根据本申请另一个实施例的图像处理方法流程图。