基于人像分割模型生成结果的图像后处理方法技术

本发明专利技术公开了基于人像分割模型生成结果的图像后处理方法，将原始图片压缩后输入人像分割模型，根据人像分割模型生成结果将图片划分为背景区、待确定区和前景区；对待确定区采用颜色梯度分析，确定待确定区每一个像素属于前景区还是背景区；将图片放大至与原始图片大小匹配，再进行二值化与滤波，得到处理后的图片数据。本发明专利技术采用图像后处理技术，解决小模型处理结果精度不佳及放大存在锯齿感的问题，使得硬件性能不够强的设备通过小模型加模型后处理得到的结果达到与硬件性能较强的设备运行的较复杂的模型得到的结果在表现上相仿的程度，降低人像抠图对硬件的要求。降低人像抠图对硬件的要求。降低人像抠图对硬件的要求。

【技术实现步骤摘要】
基于人像分割模型生成结果的图像后处理方法


[0001]本专利技术涉及图像处理
，具体的说，是一种基于人像分割模型生成结果的图像后处理方法。

技术介绍

[0002]抠图是指准确提取静止图片或者视频图片序列中的前景目标，它是许多图像编辑中的关键技术之一。抠图问题也可以被定义为每一个图片像素估计alpha值(前背景色比例)的任务。在该任务中，输入为原图像，输出将是每个像素点的alpha值。目前抠图问题往往通过深度学习模型直接得到结果即每一个图片像素的alpha值，模型复杂度决定着抠图结果的好坏，通常较好的抠图结果往往需要较复杂的模型。然而要达到实时抠图的话，模型越复杂对设备的计算能力要求就越高，往往在有限的计算能力的情况下无法部署足够复杂的模型。而如果部署较为简单的模型，虽能满足实时处理的需求，得到的人像抠图结果只能得到一个准确度较低，且输出放大后会有锯齿感的模型。还有，深度学习模型对输入数据与输出数据的大小有要求，如果输入输出数据的太大，那么模型运行时间会变长，而如果输入输出数据的大小，虽然模型运行时间会缩短，但是匹配到原图像之后会存在锯齿感，影响精度。

技术实现思路

[0003]本专利技术的目的在于提供一种基于人像分割模型生成结果的图像后处理方法，用于解决现有技术中深度学习模型处理人像抠图问题存在处理时间与模型精度的矛盾的问题，以及解决现有技术中输入输出数据大小影响模型运行时间或者导致精度降低的问题。
[0004]本专利技术通过下述技术方案解决上述问题：
[0005]一种基于人像分割模型生成结果的图像后处理方法，包括：
[0006]步骤S100、将原始图片压缩后输入人像分割模型，根据人像分割模型生成结果将图片划分为背景区、待确定区和前景区；
[0007]步骤S200、对待确定区采用颜色梯度分析，确定待确定区每一个像素属于前景区还是背景区；
[0008]步骤S300、将图片放大至与原始图片大小匹配，再进行二值化与滤波，得到处理后的图片数据。
[0009]对待确定区采用颜色梯度分析，确定待确定区每一个像素属于前景区还是背景区的方法为：遍历待确定区的每个点，执行以下步骤：
[0010]步骤S210、以选中的点I(x,y)为原点，生成检测框，检测框的左下角坐标(x
‑
5,y
‑
5)与右上角(x+5,y+5)；
[0011]步骤S220、统计检测框内所有点的r，g，b的平均值得到r测，g测，b测；
[0012]步骤S230、以点I(x,y)做十字线，找到十字线与确定区域最近的两个交点，确定区域为前景区和背景区，判断这两个交点：
[0013]若这两个交点均在前景区，则点I(x,y)在前景区，结束；
[0014]若这两个交点均在背景区，则点I(x,y)在背景区，结束；
[0015]若一个交点在前景区即前景交点，一个交点在背景区即背景交点，则分别以前景交点、背景交点为原点，生成与步骤S210中同样大小的两个检测框，分别计算两个检测框内所有点的r，g，b的平均值，得到r前，g前，b前和r背，g背，b背；比较[(r前
‑
r测)^2+(g前
‑
g测)^2+(b前
‑
b测)^2]与[(r背
‑
r测)^2+(g背
‑
g测)^2+(b背
‑
b测)^2]，若[(r前
‑
r测)^2+(g前
‑
g测)^2+(b前
‑
b测)^2]小于[(r背
‑
r测)^2+(g背
‑
g测)^2+(b背
‑
b测)^2]，则点I(x,y)在前景区，若[(r前
‑
r测)^2+(g前
‑
g测)^2+(b前
‑
b测)^2]大于[(r背
‑
r测)^2+(g背
‑
g测)^2+(b背
‑
b测)^2]，则点I(x,y)在背景区；根据[(r前
‑
r测)^2+(g前
‑
g测)^2+(b前
‑
b测)^2]、[(r背
‑
r测)^2+(g背
‑
g测)^2+(b背
‑
b测)^2]的计算结果对点I(x,y)在人像分割模型的生成结果进行加权，结束。
[0016]出于准确性的考量，我们用结合颜色梯度分析结果对模型生成结果进行加权，既对颜色梯度分析结果中得到为背景结果的在模型生成结果中进行减小操作使其趋近0，而对颜色梯度分析结果中得到为前景结果的在模型生成结果中进行加大操作使其趋近1。这一步的操作有助于结果更趋向0与1，从而减小二分划后的误差。
[0017]所述步骤S300具体包括：
[0018]将人像分割模型的生成结果更新为(255
‑
255*生成结果)，得到alpha的值，采用插值方法INTER_CUBIC将图片放大至与原始图片大小匹配，再进行二值化和高频滤波，得到图片压缩前的图片数据。
[0019]如果需要放大操作，则需要把放大操作放在二值化操作之前。原因是如果二值化后再放大会损失更多的边缘细节，会导致结果锯齿感非常明显。而关于放大操作是因为，为了保证模型速度，模型的输入输出的大小通常都会进行压缩，需要对生成结果进行放大从而匹配原始图片。二值化操作则可根据情况，在0～1之间任取一值α，而α越趋近于1，前景缺少的可能性会增大，但背景混入前景的可能性减小，α越趋近于0，前景缺少的可能性会减小，但背景混入前景的可能性增大。因此α的取值需根据模型及场景进行调整，才可找到最合适的值。二值化后，再将生成结果转换为(255
‑
255*生成结果)，得到的结果即为该像素对应的透明度alpha。
[0020]最后将上一步的透明度进行高频滤波，因此边缘锯齿说到底也就是一个高频信号，对高频信号进行滤波即可减少锯齿感的影响。滤波器的大小可以根据时间要求，图片大小以及效果进行自主调节。
[0021]本专利技术与现有技术相比，具有以下优点及有益效果：
[0022](1)本专利技术采用图像后处理技术，解决小模型处理结果精度不佳及放大存在锯齿感的问题，使得在硬件性能不够强的设备通过小模型以及模型后处理得到的结果，与在硬件性能较强的设备运行的较复杂的模型得到的结果效果相同或接近，降低人像抠图对硬件的要求。
[0023](2)本专利技术保留了小模型的速度优势，同时运用了相关的图像后处理技术使其精度得到进一步提升，以及放大后消除了抠图锯齿感的影响。
[0024](3)本专利技术采用传统图像处理方法来取代增加深度学习模型复杂度的情况，和现有技术相比，可以在低性能机器上对低精度结果进行处理，使其放大后锯齿感减少以及精
度提升。
附图说明
[0025]图1为本专利技术的流程图。
具体实施方式...

【技术保护点】

【技术特征摘要】
1.一种基于人像分割模型生成结果的图像后处理方法，其特征在于，包括：步骤S100、将原始图片压缩后输入人像分割模型，根据人像分割模型生成结果将图片划分为背景区、待确定区和前景区；步骤S200、对待确定区采用颜色梯度分析，确定待确定区每一个像素属于前景区还是背景区；步骤S300、将图片放大至与原始图片大小匹配，再进行二值化与滤波，得到处理后的图片数据。2.根据权利要求1所述的基于人像分割模型生成结果的图像后处理方法，其特征在于，对待确定区采用颜色梯度分析，确定待确定区每一个像素属于前景区还是背景区的方法为：遍历待确定区的每个点，执行以下步骤：步骤S210、以选中的点I(x,y)为原点，生成检测框，检测框的左下角坐标(x
‑
5,y
‑
5)与右上角(x+5,y+5)；步骤S220、统计检测框内所有点的r，g，b的平均值得到r测，g测，b测；步骤S230、以点I(x,y)做十字线，找到十字线与确定区域最近的两个交点，确定区域为前景区和背景区，判断这两个交点：若这两个交点均在前景区，则点I(x,y)在前景区，结束；若这两个交点均在背景区，则点I(x,y)在背景区，结束；若一个交点在前景区即前景交点，一个交点在背景区即背景交点，则分别以前景交点、背景交点为原点，生成与步骤S210中同样大小的两个检测框，分别计算两个检测框内所有点的r，g，b的平均值，得到r前，g前，b前和r背，g背，b背；比较[(r前
‑
r测)^2+(g前
‑
g测)^2+(b前
‑
b测)^2]与[(r背
‑
r测)^2+(...

【专利技术属性】
技术研发人员：李家乾，
申请(专利权)人：四川长虹电器股份有限公司，
类型：发明
国别省市：