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

说明书

本申请公开了一种图像处理方法、装置、电子设备及存储介质，该图像处理方法包括：将第一分辨率的第一图像处理为第二分辨率的第二图像，其中，所述第二分辨率小于所述第一分辨率；将所述第二图像输入至预先训练的阴影检测模型，得到所述阴影检测模型输出的阴影区域对应的掩膜图像，其中，所述掩膜图像中阴影区域的像素值与除所述阴影区域以外的其他区域的像素值不同；将所述掩膜图像以及所述第二图像输入至阴影去除模型，获得对所述第二图像去除阴影后的第三图像；根据所述第三图像以及所述掩膜图像，对所述第一图像进行阴影去除，获得去除阴影后的所述第一分辨率的第四图像。本方法可以实现对高分辨率图像的阴影去除的较佳效果。

技术领域

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

背景技术

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

发明内容

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

第一方面，本申请实施例提供了一种图像处理方法，所述方法包括：将第一分辨率的第一图像处理为第二分辨率的第二图像，其中，所述第二分辨率小于所述第一分辨率；将所述第二图像输入至预先训练的阴影检测模型，得到所述阴影检测模型输出的阴影区域对应的掩膜图像，其中，所述掩膜图像中阴影区域的像素值与除所述阴影区域以外的其他区域的像素值不同；将所述掩膜图像以及所述第二图像输入至阴影去除模型，获得对所述第二图像去除阴影后的第三图像；根据所述第三图像以及所述掩膜图像，对所述第一图像进行阴影去除，获得去除阴影后的所述第一分辨率的第四图像。

第二方面，本申请实施例提供了一种一种图像处理装置，所述装置包括：分辨率调整模块、阴影检测模块、第一阴影去除模块以及第二阴影去除模块，其中，所述分辨率调整模块用于将第一分辨率的第一图像处理为第二分辨率的第二图像，其中，所述第二分辨率小于所述第一分辨率；所述阴影检测模块用于将所述第二图像输入至预先训练的阴影检测模型，得到所述阴影检测模型输出的阴影区域对应的掩膜图像，其中，所述掩膜图像中阴影区域的像素值与除所述阴影区域以外的其他区域的像素值不同；所述第一阴影去除模块用于将所述掩膜图像以及所述第二图像输入至阴影去除模型，获得对所述第二图像去除阴影后的第三图像；所述第二阴影去除模块用于根据所述第三图像以及所述掩膜图像，对所述第一图像进行阴影去除，获得去除阴影后的所述第一分辨率的第四图像。

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

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

本申请提供的方案，通过将第一分辨率的第一图像处理为第二分辨率的第二图像，该第二分辨率小于该第一分辨率，将第二图像输入至预先训练的阴影检测模型，得到阴影检测模型输出的阴影区域对应的掩膜图像，该掩膜图像中阴影区域的像素值与除阴影区域以外的其他区域的像素值不同，再将掩膜图像以及第二图像输入至阴影去除模型，获得对第二图像去除阴影后的第三图像，然后根据第三图像以及掩膜图像，对第一图像进行阴影去除，获得去除阴影后的第一分辨率的第四图像。从而可以实现在对较高分辨率的图像进行阴影去除时，转换为低分辨率图像进行阴影去除，提升处理效率，再参考去除阴影后的低分辨率图像，对较高分辨率的图像进行阴影去除，从而可以在保证处理效率的情况下，提升阴影去除的图像的质量。

附图说明