[发明专利]图像处理方法及装置

权利要求书

1.一种图像处理方法，其特征在于，包括：

在第一图像中存在阴影区域的情况下，将所述第一图像输入至预设模型；

利用所述预设模型对所述阴影区域进行颜色矫正，得到第二图像；

其中，所述预设模型的训练停止条件为训练样本图像的预测无阴影图像与所述训练样本图像的标签无阴影图像的颜色损失值满足预设条件。

2.根据权利要求1所述的方法，其特征在于，所述在第一图像中存在阴影区域的情况下，将所述第一图像输入至预设模型之前，还包括：

计算所述第一图像中的阴影区域在所述第一图像中的占比；

在所述占比大于或等于预设占比的情况下，确定所述第一图像中存在阴影区域。

3.根据权利要求2所述的方法，其特征在于，所述计算所述第一图像中的阴影区域在所述第一图像中的占比，包括：

对所述第一图像进行高斯滤波处理，得到第三图像；所述第三图像用于反映所述第一图像的光照信息；

对所述第三图像进行二值化处理，得到第四图像；

对所述第四图像进行连通成分分析，计算所述第四图像中的连通成分的面积；

计算所述连通成分的面积在所述第一图像的全图面积中的占比。

4.根据权利要求3所述的方法，其特征在于，所述对所述第一图像进行高斯滤波处理，得到第三图像，包括：

对所述第一图像进行缩放处理，得到第五图像；所述第五图像的面积小于或等于第一预设面积；

对所述第五图像进行高斯滤波处理，得到第三图像。

5.根据权利要求3或4所述的方法，其特征在于，所述对所述第四图像进行连通成分分析，计算所述第四图像中的连通成分的面积，包括：

对所述第四图像进行连通成分分析，得到所述第四图像中的M个连通区域；M为正整数；

去除所述M个连通区域中的第一连通区域；所述第一连通区域为面积小于第二预设面积的连通区域；

计算所述M个连通区域中的第二连通区域的总面积；所述第二连通区域为所述M个连通区域中除所述第一连通区域外的连通区域；

所述计算所述连通成分的面积在所述第一图像的全图面积中的占比，包括：

计算所述第二连通区域的总面积在所述第一图像的全图面积中的占比。

6.根据权利要求1-5任一项所述的方法，其特征在于，所述利用所述预设模型对所述阴影区域进行颜色矫正，得到第二图像之后，还包括：

计算所述第二图像的清晰度值；

在所述第二图像的清晰度值大于或等于预设清晰度值的情况下，保存所述第二图像。

7.一种图像处理装置，其特征在于，包括：

第一输入模块，用于在第一图像中存在阴影区域的情况下，将所述第一图像输入至预设模型；

矫正模块，用于利用所述预设模型对所述阴影区域进行颜色矫正，得到第二图像；

其中，所述预设模型的训练停止条件为训练样本图像的预测无阴影图像与所述训练样本图像的标签无阴影图像的颜色损失值满足预设条件。

8.根据权利要求7所述的装置，其特征在于，所述图像处理装置，还包括：

第一计算模块，用于计算所述第一图像中的阴影区域在所述第一图像中的占比；

确定模块，用于在所述占比大于或等于预设占比的情况下，确定所述第一图像中存在阴影区域。

9.根据权利要求8所述的装置，其特征在于，所述第一计算模块，包括：

滤波单元，用于对所述第一图像进行高斯滤波处理，得到第三图像；所述第三图像用于反映所述第一图像的光照信息；

二值化单元，用于对所述第三图像进行二值化处理，得到第四图像；

第一计算单元，用于对所述第四图像进行连通成分分析，计算所述第四图像中的连通成分的面积；

第二计算单元，用于计算所述连通成分的面积在所述第一图像的全图面积中的占比。