[发明专利]图像去污方法、装置、可读存储介质及计算机设备

说明书

本发明提供了一种图像去污方法、装置、可读存储介质及计算机设备，该方法包括：获取原始图像，并将所述原始图像的尺寸调整至阈值尺寸，得到第一处理图像；利用训练后的生成式对抗网络模型对所述第一处理图像进行去污处理，得到第二处理图像；将所述第一处理图像和所述第二处理图像进行结构相似性计算，得到所述第一处理图像和所述第二处理图像之间的差异特征图；比对所述差异特征图和所述第一处理图像，以确定所述第一处理图像中的污渍区域和所述污渍区域所在的位置信息；将所述污渍区域缩放至与所述原始图像对应的尺寸，并根据缩放后的污渍区域以及所述位置信息确定所述原始图像中的目标污渍区域；去除所述原始图像中的目标污渍区域的内容。

技术领域

本发明涉及电子技术领域，特别是涉及一种图像去污方法、装置、可读存储介质及计算机设备。

背景技术

在档案数字化存档、教育等领域中经常需要对纸质文档的内容进行重复印刷，一般需要通过扫描和复印方式进行。但是若纸质文档表面存不干净或者表面不平整、粗糙时，文档扫描或复印后出现很多的污渍区域（如大块的黑斑、聚集散点等），影响使用。因此，通常需要对图像进行去污处理。

传统的扫描图像去除污渍方法是对二值化后的图像进行距离变换，得到若干个图像块后，再进行连通区域分析得到包围图像块的矩形框，对矩形框内的图像块逐一分析其几何特征（如投影面积比、实际像素占矩形框面积比、轮廓、Hog特征等），对特征设定预设值，在预设值范围内的图像块视为污渍，并将该图像块的污渍标注出来，在预设值之外的图像块则视为正常图像。

由于大块污渍的形状、大小和表现形式各种各样，几何特征值无法覆盖绝大部分污渍类型，且稍微模糊的图像中，文字的特征易与污渍几何特征混淆导致文字被误去除。因而目前的污渍去除方案效果并不好，对污渍的去除没有泛化性。

发明内容

鉴于上述状况，有必要针对现有技术中图像污渍处理困难的问题，提供一种图像去污方法、装置、可读存储介质及计算机设备。

一种图像去污方法，包括：

获取原始图像，并将所述原始图像的尺寸调整至阈值尺寸，得到第一处理图像；

利用训练后的生成式对抗网络模型对所述第一处理图像进行去污处理，得到第二处理图像；

将所述第一处理图像和所述第二处理图像进行结构相似性计算，得到所述第一处理图像和所述第二处理图像之间的差异特征图；

比对所述差异特征图和所述第一处理图像，以确定所述第一处理图像中的污渍区域和所述污渍区域所在的位置信息；

将所述污渍区域缩放至与所述原始图像对应的尺寸，并根据缩放后的污渍区域以及所述位置信息确定所述原始图像中的目标污渍区域；

去除所述原始图像中的目标污渍区域的内容。

进一步的，上述图像去污方法，其中，所述比对所述差异特征图和所述第一处理图像，以确定所述第一处理图像中的污渍区域和所述污渍区域所在的位置信息的步骤包括：

对所述差异特征图进行二值化和连通区域分析处理；

比对处理后的所述差异特征图和所述第一处理图像，以确定所述第一处理图像中的差异区域；

对所述差异区域进行外轮廓提取和拟合多边形处理得到所述第一处理图像中的污渍区域，并获取所述污渍区域的位置信息。

进一步的，上述图像去污方法，其中，所述利用训练后的生成式对抗网络模型对所述第一处理图像进行去污处理，得到第二处理图像的步骤之前还包括：

利用生成模型和判别模型构建生成式对抗网络模型，利用训练组图像对所述生成式对抗网络模型进行训练，所述训练组图像包括多组内容相同的干净图像和带污渍图像。