一种基于神经网络的文件空位内容检测方法及设备技术

技术编号:35745990 阅读:10 留言:0更新日期:2022-11-26 18:50
本发明专利技术涉及一种基于神经网络的文件空位内容检测方法,包括以下步骤:采集待检测文本图像,进行第一预处理后,得到第一数据集,并且将第一数据集分为第一训练集和第一测试集;将第一训练集输入到YOLO V3神经网络中进行训练,生成训练好的YOLO V3神经网络检测模型;执行YOLO V3神经网络检测模型的应用。本发明专利技术通过采集待检测文本图像形成训练集数据,以完成YOLO V3神经网络的训练,通过使用YOLO V3神经网络检测模型完成对文本图像空位的检测,其文本空位识别准确率高。本空位识别准确率高。本空位识别准确率高。

【技术实现步骤摘要】
一种基于神经网络的文件空位内容检测方法及设备


[0001]本专利技术属于视觉识别
,特别涉及一种基于神经网络的文件空位内容检测方法及设备。

技术介绍

[0002]随着我国越来越重视行业安全问题,多数制造企业会采用工作票等形式对过往操作进行记录,以便提高员工安全意识,提高生产安全性。此类文件在工作时须有负责人逐项签字,但此类文件往往条例过多且需人工签字部分较多,所以会出现漏签等问题,因此需要相关负责人定期对过往文件进行检查。而此类文件存在文件量大,单个文件所需检查点过多等问题,占用大量工作时间,对于相关人员工作造成困扰。

技术实现思路

[0003]本专利技术的目的在于提供一种基于神经网络的文件空位内容检测方法及设备,能够实现文件空位内容检测。
[0004]为达到上述目的,本专利技术采用的技术方案是:
[0005]一种基于神经网络的文件空位内容检测方法,包括以下步骤:
[0006]采集待检测文本图像,进行第一预处理后,得到第一数据集,并且将第一数据集分为第一训练集和第一测试集;
[0007]将第一训练集输入到YOLO V3神经网络中进行训练,生成训练好的YOLO V3神经网络检测模型;
[0008]执行YOLO V3神经网络检测模型的应用;
[0009]其中,第一训练集输入到YOLO V3神经网络中进行训练步骤还包括以下步骤:
[0010]将第一训练集按乱序送入YOLO V3神经网络中,计算第一训练集的损失函数值,并且通过优化算法完成YOLO V3神经网络权重的更新;
[0011]以第一训练集输入到YOLO V3神经网络,完成权重更新为一个批次,每个批次结束之后将第一测试集送入批次更新后的YOLO V3神经网络,计算当前权重下的损失函数值,若当前批次第一测试集损失函数值低于前一次批次中的第一训练集损失函数值,则保存当前YOLO V3神经网络权重,生成YOLO V3神经网络检测模型;
[0012]经过若干批次的迭代后,选取更新后的YOLO V3神经网络检测模型。
[0013]优选的,还包括以下步骤:
[0014]采集打印数字与手写数字图像,进行第二预处理后,得到第二数据集,并且对第二数据集分为第二训练集和第二测试集;
[0015]将第二训练集输入到ResNet18神经网络中进行训练,生成训练好的ResNet18神经网络检测模型;
[0016]执行ResNet18神经网络检测模型的应用;
[0017]其中,第二训练集输入到ResNet18神经网络中进行训练步骤还包括以下步骤:
[0018]将第二训练集按乱序送入ResNet18神经网络中,计算第二训练集的损失函数值,并且通过优化算法完成ResNet18神经网络权重的更新;
[0019]以第二训练集送入ResNet18神经网络,完成权重更新为一个批次,每个批次结束之后将第二测试集送入批次更新后的ResNet18神经网络,计算当前权重下的损失函数值,若当前批次第二测试集损失函数值低于前一次批次中的第二训练集损失函数值,则保存当前Res Net18神经网络权重,生成ResNet18神经网络检测模型;
[0020]经过若干批次的迭代后,选取更新后的ResNet18神经网络检测模型。
[0021]优选的,所述第一预处理包括以下步骤:
[0022]筛选所拍摄图像,将模糊失真图像剔除;
[0023]将筛选剩余图像进行双线性插值处理,降低图像分辨率,并统一图像分辨率;
[0024]对剩余图像中的已填写的文件空位和漏填空位分别标注.
[0025]优选的,所述第二预处理包括以下步骤:
[0026]将所采集图像通过双线性插值的方式统一分辨率;
[0027]将统一分辨率之后的图像进行灰度化与二值化处理,剔除对数字识别影响较小的色彩因素。
[0028]优选的,所述YOLO V3神经网络的损失函数为:
[0029][0030]其中,Y为经由YOLO V3神经网络的输出即召回率,α为平衡因子,γ为调节因子,y标识类别。
[0031]优选的,所述ResNet18神经网络的损失函数为:
[0032][0033]其中n标识ResNet18神经网络输出个数,Y
i
标识第i个输出值,标识第i个输出值的标签值;
[0034]优选的,所述执行YOLO V3神经网络检测模型的应用包括以下步骤;
[0035]采集待检测文本图像,进行第一预处理;
[0036]输入到YOLO V3神经网络检测模型中,标识待检测文本图像中手写数字与所对应的打印数字,完成文本图像空位的检测。
[0037]优选的,所述执行ResNet18神经网络检测模型的应用包括以下步骤;
[0038]将标识后的手写数字与所对应的打印数字剪切后,对其进行第二预处理;
[0039]输入到ResNet18神经网络检测模型中,检测手写数字与所对应的打印数字是否一致。
[0040]本专利技术另一方面,还提供了一种基于神经网络的文件空位内容检测设备,包括:
[0041]存储器,用于存储软件应用程序,
[0042]处理器,用于执行软件应用程序,所述软件应用程序的各程序相对应地执行上述一种基于神经网络的文件空位内容检测方法中的步骤。
[0043]本专利技术另一方面,还提供了一种计算机可读存储介质,其存储有计算机程序,该计算机程序被处理器执行时执行上述一种基于神经网络的文件空位内容检测方法的步骤。
[0044]本专利技术通过采集待检测文本图像形成训练集数据,以完成YOLO V3神经网络的训练;通过对实际文件中打印字符与手写字符进行采集,生成训练集数据,完成对ResNet18神经网络的训练,将待检测文本输入到YOLO V3神经网络检测模型中,完成对文本图像空位的检测,同时对文本中的手写数字与所对应的打印数字进行标识和剪切,输入到ResNet18神经网络检测模型中,检测手写数字与所对应的打印数字是否一致。本专利技术将文本空位检测与空位数字内容识别分开处理,提高了文本空位检测的速度与准确性。
附图说明
[0045]图1是本专利技术一种基于神经网络的文件空位内容检测方法流程结构示意图。
具体实施方式
[0046]以下结合附图和具体实施例,对本专利技术做进一步说明。
[0047]如图1所示,一种基于神经网络的文件空位内容检测方法,包括以下步骤:
[0048]S10:采集待检测文本图像,进行第一预处理后,得到第一数据集,并且将第一数据集分为第一训练集和第一测试集;
[0049]具体的,采集待检测文本图像,包括利用相机采集不同角度文本图片;
[0050]具体的,第一训练集和第一测试集的比例为8:2,以便YOLO V3神经网络的训练;
[0051]本实施例中,步骤S10中,所述第一预处理包括以下步骤:
[0052]S11:筛选所拍摄图像,本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种基于神经网络的文件空位内容检测方法,其特征在于,包括以下步骤:采集待检测文本图像,进行第一预处理后,得到第一数据集,并且将第一数据集分为第一训练集和第一测试集;将第一训练集输入到YOLO V3神经网络中进行训练,生成训练好的YOLO V3神经网络检测模型;执行YOLO V3神经网络检测模型的应用;其中,第一训练集输入到YOLO V3神经网络中进行训练步骤还包括以下步骤:将第一训练集按乱序送入YOLO V3神经网络中,计算第一训练集的损失函数值,并且通过优化算法完成YOLO V3神经网络权重的更新;以第一训练集输入到YOLO V3神经网络,完成权重更新为一个批次,每个批次结束之后将第一测试集送入批次更新后的YOLO V3神经网络,计算当前权重下的损失函数值,若当前批次第一测试集损失函数值低于前一次批次中的第一训练集损失函数值,则保存当前YOLO V3神经网络权重,生成YOLO V3神经网络检测模型;经过若干批次的迭代后,选取更新后的YOLO V3神经网络检测模型。2.根据权利要求1所述的一种基于神经网络的文件空位内容检测方法,其特征在于,还包括以下步骤:采集打印数字与手写数字图像,进行第二预处理后,得到第二数据集,并且对第二数据集分为第二训练集和第二测试集;将第二训练集输入到ResNet18神经网络中进行训练,生成训练好的ResNet18神经网络检测模型;执行ResNet18神经网络检测模型的应用;其中,第二训练集输入到ResNet18神经网络中进行训练步骤还包括以下步骤:将第二训练集按乱序送入ResNet18神经网络中,计算第二训练集的损失函数值,并且通过优化算法完成ResNet18神经网络权重的更新;以第二训练集送入ResNet18神经网络,完成权重更新为一个批次,每个批次结束之后将第二测试集送入批次更新后的ResNet18神经网络,计算当前权重下的损失函数值,若当前批次第二测试集损失函数值低于前一次批次中的第二训练集损失函数值,则保存当前ResNet18神经网络权重,生成ResNet18神经网络检测模型;经过若干批次的迭代后,选取更新后的ResNet18神经网络检测模型。3.根据权利要求1所述的一种基于神经网络的文件空位...

【专利技术属性】
技术研发人员:杨立旋郝海军李锦乾智刚荣矗嗣
申请(专利权)人:山西昱光发电有限责任公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1