本技术涉及图像处理领域,尤其涉及一种图像处理方法、装置、设备及存储介质。
背景技术:
1、抠图是图像处理领域中的关键技术,影视行业的特效、证件照背景颜色的替换和宣传海报中都有使用到图像抠图这一技术,这使得抠图的效率和准确性格外重要。
2、现有技术都是在原图的整体上做的数据增强(比如对样本集增加噪声、色彩抖动、图像均衡化和滤波处理),这使得现有技术对毛发细节以及人像前景和背景色有明显轮廓的图片的抠图效果是很出色的,但由于不能准确地识别图像前景和背景颜色接近的场景下的边缘轮廓,导致当前景和背景颜色接近时,现有抠图技术不能准确识别图片的前景和背景。
3、上述内容仅用于辅助理解本技术的技术方案,并不代表承认上述内容是现有技术。
技术实现思路
1、本技术的主要目的在于提供一种图像处理方法、装置、设备及存储介质,旨在解决现有技术中当前景和背景颜色接近时,不能准确识别图片的前景和背景的技术问题。
2、为实现上述目的,本技术提供一种图像处理方法,所述图像处理方法包括以下步骤:
3、在接收到待处理图像时,确定待处理图像的前景区域;
4、在所述待处理图像的前景区域和背景区域的区分度小于第一预设值时,计算所述待处理图像的前景区域的三通道色阶值,并基于所述三通道色阶值修改所述待处理图像的前景区域的色阶,以得到最终的抠图结果,其中,修改色阶后的待处理图像的前景区域和背景区域的区分度大于第二预设值,第二预设值必须大于第一预设值。
5、可选地,所述在接收到待处理图像时,确定待处理图像的前景区域的步骤,包括:
6、在接收到待处理图像时,将所述待处理图像输入至预设的抠图模型,其中,所述抠图模型是基于数据训练集中原图对应的目标掩码标签图,对预设的待训练抠图模型进行迭代训练得到的,所述目标掩码标签图用于表征数据训练集中原图的前景区域;
7、基于所述抠图模型,确定所述待处理图像的前景区域。
8、可选地,所述在接收到待处理图像时,将所述待处理图像输入至预设的抠图模型的步骤之前,所述方法包括:
9、获取数据训练集中图像的原图和原图所对应的目标掩码标签图;
10、基于所述原图所对应的目标掩码标签图,获取所述原图的前景区域的三通道色阶值;
11、基于所述原图的前景区域的三通道色阶值,修改所述原图的色阶,得到色阶修改图,以实现所述数据训练集的数据扩充;
12、将数据扩充后的数据训练集输入至所述预设的待训练抠图模型,得到图像选择信息;
13、将所述图像选择信息与所述数据训练集中原图所对应的目标掩码标签图进行差异计算,得到第一损失值;
14、判断所述第一损失值是否满足预设损失阈值范围指示的误差标准;
15、若所述第一损失值未满足预设损失阈值范围指示的误差标准,则返回将数据扩充后的数据训练集输入至所述预设的待训练抠图模型,得到图像选择信息的步骤,直到所述第一损失值满足预设损失阈值范围指示的误差标准后停止训练,得到满足精度条件的抠图模型。
16、可选地,所述在所述待处理图像的前景区域和背景区域的区分度小于第一预设值时,计算所述待处理图像的前景区域的三通道色阶值,并基于所述三通道色阶值修改所述待处理图像的前景区域的色阶,以得到最终的抠图结果的步骤,包括:
17、在所述待处理图像的前景区域和背景区域的区分度小于第一预设值时,计算所述待处理图像的前景区域的三通道色阶值;
18、基于所述待处理图像的前景区域的三通道色阶值,修改所述待处理图像的前景区域的色阶,得到色阶处理图;
19、将所述色阶处理图输入至所述抠图模型,得到色阶处理图的前景区域;
20、将所述待处理图像的前景区域与所述色阶处理图的前景区域合并,得到最终的抠图结果。
21、可选地,所述在所述待处理图像的前景区域和背景区域的区分度小于第一预设值时,计算所述待处理图像的前景区域的三通道色阶值的步骤,包括:
22、在所述待处理图像的前景区域和背景区域的区分度小于第一预设值时,获取所述待处理图像的前景区域中每个色阶单元的数值;
23、基于所述每个色阶单元的数值计算出每个色阶通道的色阶数据;
24、将每个色阶通道的色阶数据结合,得到所述待处理图像的前景区域的三通道色阶值。
25、可选地,所述基于所述待处理图像的前景区域的三通道色阶值,修改所述待处理图像的色阶,得到色阶处理图的步骤,包括:
26、基于所述待处理图像的前景区域的三通道色阶值,确定色阶的调整比例;
27、基于所述色阶的调整比例,修改所述待处理图像的色阶,得到色阶处理图。
28、可选地,所述将所述待处理图像的前景区域与所述色阶处理图的前景区域合并,得到最终的抠图结果的步骤,包括:
29、基于所述待处理图像的前景区域,获取所述待处理图像的前景区域的细节部分;
30、基于所述色阶处理图的前景区域,获取所述色阶处理图的前景区域的轮廓部分;
31、将所述待处理图像的前景区域的细节部分和所述色阶处理图的前景区域的轮廓部分合并,得到最终的抠图结果。
32、此外,为实现上述目的,本技术还提供一种图像处理装置,所述装置包括:
33、抠图模块,用于在接收到待处理图像时,确定待处理图像的前景区域;
34、处理模块,用于在所述待处理图像的前景区域和背景区域的区分度小于第一预设值时,计算所述待处理图像的前景区域的三通道色阶值,并基于所述三通道色阶值修改所述待处理图像的前景区域的色阶,以得到最终的抠图结果,其中,修改色阶后的待处理图像的前景区域和背景区域的区分度大于第二预设值,第二预设值必须大于第一预设值。
35、此外,为实现上述目的,本技术还提供一种图像处理设备,所述设备包括:存储器、处理器及存储在所述存储器上并可在所述处理器上运行的图像处理程序,所述图像处理程序配置为实现如上所述的图像处理方法的步骤。
36、此外,为实现上述目的,本技术还提供一种存储介质,所述存储介质上存储有图像处理程序,所述图像处理程序被处理器执行时实现如上所述的图像处理方法的步骤。
37、本技术提供一种图像处理方法、装置、设备及存储介质,与相关技术中在原图的整体上做的数据增强(比如对样本集增加噪声、色彩抖动、图像均衡化和滤波处理),导致当前景和背景颜色接近时,不能准确识别图片的前景和背景相比,本技术通过在接收到待处理图像时,确定待处理图像的前景区域,然后在所述待处理图像的前景区域和背景区域的区分度小于第一预设值时,计算所述待处理图像的前景区域的三通道色阶值,并基于所述三通道色阶值修改所述待处理图像的前景区域的色阶,以得到最终的抠图结果,由于本技术计算所述待处理图像的前景区域的三通道色阶值,并可以对色阶进行修改,且色阶修改后的所述待处理图像的前景区域和背景区域的颜色差不接近,使得本技术在图片前景和背景颜色接近的场景下,可以准确识别图片的前景和背景。