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

权利要求书

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

获取目标图像，所述目标图像包括待检测的光源对象；

检测所述目标图像中光源对象的位置，得到所述光源对象的位置信息；

对所述位置信息指示的光源图像进行去光晕处理，得到去光晕后的光源图像；

对所述去光晕后的光源图像进行描色处理，得到处理后的目标图像，所述处理后的目标图像包括处理后的光源图像；

所述检测所述目标图像中光源对象的位置，包括：

获取目标检测模型，所述目标检测模型包括第一网络模型和第二网络模型，所述第一网络模型用于对所述光源对象的位置进行预测；所述第二网络模型用于基于所述第一网络模型中至少一层的输出结果使用不同尺寸特征图对所述光源对象的位置进行预测；

将所述目标图像输入预设的目标检测模型，得到所述光源对象的位置。

2.根据权利要求1所述的方法，其特征在于，所述获取目标检测模型之前，还包括：

将样本数据输入初始网络模型，所述初始网络模型的模型结构与所述目标检测模型的模型结构相同；

基于分类损失函数和回归损失函数对所述初始网络模型进行训练，得到所述目标检测模型。

3.根据权利要求1所述的方法，其特征在于，所述第一网络模型包括SSD网络模型，所述第二网络模型包括特征图金字塔网络。

4.根据权利要求1所述的方法，其特征在于，所述对所述位置信息指示的光源图像进行去光晕处理，得到去光晕后的光源图像，包括：

获取去光晕算法，所述去光晕算法用于均衡所述光源图像中各个位置的颜色值；

将所述位置信息输入所述去光晕算法，得到所述去光晕后的光源图像。

5.根据权利要求4所述的方法，其特征在于，所述去光晕算法通过下式表示：

f(x)＝-g(x)×x_RGB

其中，x _RGB为所述光源图像中像素x处的RGB值、f(x)为去光晕后的光源图像中像素x处的RGB值、g(x)为像素x处RGB值的调节系数、μ为所述光源图像的中心位置，σ为所述光源图像中各个像素点的RGB值的标准差。

6.根据权利要求1所述的方法，其特征在于，所述对所述去光晕后的光源图像进行描色处理，得到处理后的目标图像，包括：

确定与所述去光晕后的光源图像对应的模板图像，所述模板图像的类型与所述光源对象的类型相同；

基于所述去光晕后的光源图像中指定位置处的像素值确定所述模板图像的模板底色；

使用具有所述模板底色的模板图像对所述去光晕后的光源图像进行描色。

7.根据权利要求1至6任一所述的方法，其特征在于，所述方法还包括：

获取所述光源对象的对象信息；

基于所述对象信息对所述光源对象的位置信息进行调整。

8.根据权利要求1至6任一所述的方法，其特征在于，所述方法还包括：

确定当前位置的HSV颜色空间是否为指定颜色的HSV颜色空间；

在所述当前位置的HSV颜色空间是所述指定颜色的HSV颜色空间时，触发执行所述对所述位置信息指示的光源图像进行去光晕处理，从而得到去光晕后的光源图像。