一种基于扫描图像自动黑边检测修复的方法技术

本发明专利技术公开了一种基于扫描图像自动黑边检测修复的方法，包括：图像大小处理、预设检测位置、图像黑点区域填充、黑边区域数据扩大和消噪、掩码归一化、生成制作数据集、模型训练、模型推理、计算修复区域、区域尺寸还原、合成修复图像。本发明专利技术的有益效果是图像满足修复功能，实现了黑色区域预取、黑色区域图像修复生成、图像合成、边界自然高清、快速修复的功能。快速修复的功能。快速修复的功能。

【技术实现步骤摘要】
一种基于扫描图像自动黑边检测修复的方法


[0001]本专利技术涉及扫描文档电子化
，具体涉及一种基于扫描图像自动黑边检测修复的方法。

技术介绍

[0002]在纸质文件电子化过程中，因为纸质文档倾斜或纸质文档四周破损缺失，扫描后的图像存在四周黑边、缺失数据区域，呈现不规则外围黑色区域情况。常见方法可以实现规则黑边去除、不规则黑边去除，对于黑边进行简单的去除处理后，图像页面会形成明显的不协调区域，即正常数据区域和黑边区域程序边界明显，黑边区域没有得到数据填充修复。

技术实现思路

[0003]针对上述问题，本方法的目的是对扫描图像自动提取黑色区域，利用深度学习训练、推理生成与对应黑色周边区域和文档相协调的新数据区域，然后合成完整的修复图像，可以在应用数字档案系统中大量修复数字档案使用。本专利技术提供一种基于扫描图像自动黑边检测修复的方法,
[0004]本专利技术采用下述的技术方案：
[0005]一种基于扫描图像自动黑边检测修复的方法，包括以下步骤：
[0006]S1、获取原扫描图像生成第一图像和第二图像；
[0007]S2、对所述第一图像进行大小处理；
[0008]S3、在所述第二图像四角位置分别预设4个检测位置；
[0009]S4、判断所述4个检测位置的第一维像数值是否小于阈值，当检测位置小于阈值，在原扫描图像进行满水填充，即取得带有黑边区域的掩码图像；
[0010]S5、对所述掩码图像进行消噪和归一化处理；
[0011]S6、重复步骤S2至S5得到黑边修复文件，生成制作数据集，基于AOT
‑
GAN网络进行图像生成训练，生成修复模型；
[0012]S7、将所述第一图像输入进训练后的AOT
‑
GAN，重复步骤S2至S5，生成掩码数据，第一图像根据掩码数据推理计算生成第三图像；
[0013]S8、所述第三图像与步骤S7所述掩码数据进行掩码计算，提取得到只有掩码区域的修复图像第四图像，并对所述第四图像进行图像大小处理；
[0014]S9、将第四图像与原扫描图像合成得到修复图像。
[0015]进一步的，所述原扫描图像为RGB图像。
[0016]进一步的，所述第一图像大小为512*512。
[0017]进一步的，在步骤S6中，对所述AOT
‑
GAN模型训练进行改进，具体包括：
[0018]S61、将原始的[1、2、4、8]四个尺度的空洞卷积替换为[2，4，8、16]四个尺度的空洞卷积；
[0019]S62、用GELU激活函数替换之前的RELU激活函数。
[0020]本专利技术的有益效果是：
[0021]图像满足修复功能，实现了黑色区域预取、黑色区域图像修复生成、图像合成、边界自然高清、快速修复的功能。
附图说明
[0022]为了更清楚地说明本专利技术实施例的技术方案，下面将对实施例的附图作简单地介绍，显而易见地，下面描述中的附图仅仅涉及本专利技术的一些实施例，而非对本专利技术的限制。
[0023]图1为本申请实施例提供的工作流程图；
[0024]图2为本申请实施例提供的原扫描图像；
[0025]图3为本申请实施例带有黑边区域的掩码图像示意图；
[0026]图4为本申请实施例消噪处理后的掩码图像示意图；
[0027]图5为申请实施例第四图像示意图；
[0028]图6为本申请实施例修复图像示意图
具体实施方式
[0029]为使本专利技术实施例的目的、技术方案和优点更加清楚，下面将结合本专利技术实施例的附图，对本专利技术实施例的技术方案进行清楚、完整地描述。显然，所描述的实施例是本专利技术的一部分实施例，而不是全部的实施例。基于所描述的本专利技术的实施例，本领域普通技术人员在无需创造性劳动的前提下所获得的所有其他实施例，都属于本专利技术保护的范围。
[0030]下面结合附图和实施例对本专利技术进一步说明。
[0031]如图1所示，为本申请一种基于扫描图像自动黑边检测修复的方法，具体包括以下步骤：
[0032]步骤一、获取原扫描图像生成第一图像和第二图像。
[0033]图像存在黑边的主要原因是在扫描过程中图片未放置平整或者图像出现破损卷角情况，本方法处理扫描图像为RGB图像，在图像黑边检测过程中不需要对图像做灰度或二值化处理。
[0034]如图2所示为原扫描图像，获取原扫描图像高h与宽度w,生成宽度w+2,高度h+2全0的掩码矩阵数据，即数据四周进行一圈的0填充，保障下一步填充颜色处理过程后能获取到图像四周边界完整数据。
[0035]进一步的，需要将原扫描图像生成2个图像副本分别为第一图像和第二图像。
[0036]步骤二、对所述第一图像进行大小处理。
[0037]将第一图像进行归一化大小处理，尺寸大小为512*512，因为后续深度学习修复模型AOT
‑
GAN只能处理不高于512*512大小的图像。
[0038]步骤三、在所述第二图像四角位置分别预设4个检测位置。
[0039]在第二图像四角位置分别预设四个预处理点，坐标为[3][3]、[h
‑
3][3]、[3][w
‑
3]、[h
‑
3][w
‑
3]。
[0040]步骤四、判断所述4个检测位置的第一维像数值是否小于阈值，当检测位置小于阈值，在原扫描图像进行满水填充，即取得带有黑边区域的掩码图像。
[0041]针对出现黑边情况，只需要检测这四个点的第一维像数值是否小于阈值A，A默认
值设置为4，也可以根据实际情况调整阈值A。针对小于阈值A的点，在原扫描图上进行满水填充，mask为填充数据区域，填充颜色为绿色(255，0，0)。无论四方的黑框是否相连接，通过对四个点的漫水填充，可用绿色填充图像四周所有黑边区域，对于非黑边区域，则不会进行任何填充，即不会破坏图像正常数据。
[0042]步骤五、对所述掩码图像进行消噪和归一化处理。
[0043]带有黑边区域的掩码图像为不规则图形，存在多个不连通区域，如图3所示。如果按这个区域在原图中直接去除黑色边框，可以到达去除黑边效果，但是原图中的正常图像和黑边区域之间存在边界痕迹和噪声，非常不协调。所以对当前带有黑边区域的掩码图像进行反色处理，即每一个像素从0变为255或255变为0.然后进行腐蚀操作，腐蚀操作会让正常图像区域减小、噪声图像区域减少甚至淹没消失，最后再进行反色操作，取得优化的掩码图像，如图4所示。
[0044]进一步的，对所述优化的掩码图像归一化为512*512的大小，在进行归一化大小处理之前需要将优化的掩码图像合并新的三通道图像，保持优化的掩码图像与原扫描图像的通道数目一致。
[0045]步骤六、重复步骤S2至S5得到黑边修复文件，生成制作数据集，基于AOT
‑
GAN网络进行图像生成训练，生成修复本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于扫描图像自动黑边检测修复的方法，其特征在于，包括以下几个步骤：S1、获取原扫描图像生成第一图像和第二图像；S2、对所述第一图像进行大小处理；S3、在所述第二图像四角位置分别预设4个检测位置；S4、判断所述4个检测位置的第一维像数值是否小于阈值，当检测位置小于阈值，在原扫描图像进行满水填充，即取得带有黑边区域的掩码图像；S5、对所述掩码图像进行消噪和归一化处理；S6、重复步骤S2至S5得到黑边修复文件，生成制作数据集，基于AOT
‑
GAN网络进行图像生成训练，生成修复模型；S7、将所述第一图像输入进训练后的AOT
‑
GAN，重复步骤S2至S5，生成掩码数据，第一图像根据掩码数据推理计算生成第三图像；S8、所述第三图像与步骤S7所述掩码数据进行掩码计算，提取得到只有掩码区域的修复图像第四图像，并对所述第四图像进行图像大小处理；S9、将第四图像与原扫描图像合成得到修复图像。2.根据权利要求1所述一种基于扫描图像自动黑边检测修复的方法，其特征在于，所述原扫描图像为RGB图像。3.根据权利要求1所述一种基于扫描图像自动黑边检...

【专利技术属性】
技术研发人员：王敏，陈蕊，黄莉，周杨，刘佳，唐娅梅，
申请(专利权)人：四川工商学院，
类型：发明
国别省市：