一种基于深度学习的指针式水表读数检测方法技术

技术编号:21273997 阅读:106 留言:0更新日期:2019-06-06 08:14
本发明专利技术公开了一种基于深度学习的指针式水表读数检测方法,包括S1采集含有指针式表盘的水表图片,对图像进行预处理;S2对预处理后的图像指针表盘进行标注,制作数据标签,将数据标签随机划分成训练集和测试集;S3构建基于深度学习网络的检测模型;S4使用训练集训练检测模型,将测试集输入训练好的检测模型,得到适用于指针式水表读数的检测模型。测试表明,模型平均检测准确率超过97%。本发明专利技术可实现大规模指针式水表表盘检测,有效解决表盘指针短小而难以精确检测的问题,因此具有较好的实际应用价值。

【技术实现步骤摘要】
一种基于深度学习的指针式水表读数检测方法
本专利技术涉及人工智能
,具体涉及一种基于深度学习的指针式水表读数检测方法。
技术介绍
随着人工智能技术的不断推进和大数据时代的到来,深度学习技术在文字识别、图像识别等领域突破了多个难题,以及近年来显卡等硬件技术日新月异的发展,为基于深度学习网络的进一步实际应用提供了可靠的技术基础。日常用水作为每家每户的必需品,水费收缴也成为不可或缺的工作,但水表数据记录一直是全国一个耗时又易错的难题。常见记录方式有:(1)人工抄表。这种方式常常花费大量的人力和物力,易出错且速度慢。(2)可自动发送数据的智能电子水表。目前已有部分新建小区采用,但由于这类水表的造价高昂、安装条件要求高,大部分老式楼房并不适合安装,因而难以普及,使用量极少。(3)基于图像识别技术的自动检测和识别水表图片中的读数。通过摄像头拍摄水表图像,利用传统图像识别方法或者深度学习网络进行图像检测与识别,这种方法成本较低,易于推广。但已有基于传统图像识别方法的指针式表盘检测,检测准确度容易受图像噪声的影响,精度也无法满足实际应用的需要。且已有的方法主要针对压力表、监测仪表等具有细长指针的表盘,并不适应于指针式水表的检测。而已有的基于深度学习网络的检测模型有FastR-CNN、SSD、YOLO等,虽然各有优势,但是目前所作的研究工作主要是针对数字式表盘,而对指针式表盘的检测与识别效果并不理想,难以广泛应用,其中存在的主要问题是水表中的指针式表盘区域分散,指针短小,表盘刻度小,指针在阴影等干扰因素下特征不明显等。
技术实现思路
为了克服现有技术存在的缺点与不足,本专利技术提供一种基于深度学习的指针式水表读数检测方法。本专利技术采用如下技术方案:一种基于深度学习的指针式水表读数检测方法,包括如下步骤:S1采集含有指针式表盘的水表图片,对图像进行预处理;S2对预处理后的图像指针表盘进行标注,制作数据标签,将数据标签随机划分成训练集和测试集;S3构建基于深度学习网络的检测模型;S4使用训练集训练检测模型,得到适用于指针水表读数的检测模型,并将测试集输入训练好的检测模型中,测试检测模型的性能。所述S3中基于深度学习网络的检测模型包括基于残差网络的特征提取模块、用于获取感兴趣区域的RPN模块、用于获取分数地图的分类模块及位置敏感的ROIPooling模块。所述RPN模块还包括bottleneck结构的残差块,RPN模块通过对预处理后图片每个区域与anchor的偏移量计算及区域隶属于不同类别概率计算,然后根据偏移量及概率筛选出感兴趣区域。所述分类模块包括ResNet101的conv5_x部分及多尺度特征检测部分以及一个卷基层部分,将多尺度特征检测部分的输出及ResNet101的conv5_x部分的输出作为输入给卷积层,进一步输出分数地图。所述卷积层增加一个bottleneck结构的残差块。所述位置敏感的ROIPooling模块是对感兴趣区域及分数地图进行pooling;输入为k2×(C+1)×W′×H′,其中k2为分数地图的尺寸,(C+1)为类别数,W′和H′为区域的宽和高,输出为一个为k2×(C+1)的立体块;对输出进行投票得到最后的结果,即根据每个区域所属的类别和相应的概率,得到位置敏感区域。所述anchor设置长度比例为1:1。所述水表图片中水表指针水平旋转在±30度以内。本专利技术的有益效果:(1)本专利技术可以大大缩短抄表人力和物力的投入,只需要对水表进行拍摄取样,适合大量水表图片数据的检测;(2)基于深度学习的神经网络,充分利用了深度网络自学能力,提取水表图像中的指针表盘的图像特征;(3)通过在RPN模块和分类模块的改进,有效的提高检测准确率和改善因网络过深导致的检测准确率降低的问题;(4)本专利技术的模型检测准确率较高,速度较快,鲁棒性强,不仅可以有效检测到指针式表盘,更可以根据指针位置进行分类,亦有利于进一步的指针表盘识别工作。附图说明图1是本专利技术的工作流程图;图2是本专利技术检测模型的结构示意图;图3是本专利技术改进的RPN模块示意图;图4是本专利技术改进的分类模块结构示意图;图5是本专利技术实施例中检测结果示意图。具体实施方式下面结合实施例及附图,对本专利技术作进一步地详细说明,但本专利技术的实施方式不限于此。实施例如图1所示,一种基于深度学习的指针式水表读数检测方法,包括采集预处理步骤、制作数据集、构建基于深度学习网络的检测模型、训练模型和测试模型步骤,具体为:一采集和预处理步骤用摄像头或移动终端等具有拍摄功能的设备对含有指针式表盘的水表进行图像采集,对图像进行预处理,无需修改原始水表设备,只要拍摄的图像包含水表中的指针式表盘,且水平旋转在±30度以内,正常清晰度即可。并对图片做正方形剪裁、空白处用黑色填充等基本图像处理。且不受限于其他因素,既可用于纯指针式表盘的水表读数检测,亦可用于含有数字式和指针式表盘的水表读数的检测。二制作数据集对处理后的图像指针表盘进行标注,制作数据标签,针对水表特性,数据标签将指向不同数字的指针式表盘标注为不同的对象,具体是:当水表指针指向数字“1”到指向数字“2”之间的标注为“数字1”,把指向数字“2”到指向数字“3”之间的标注为“数字2”,一共可得到11类(含背景),有利于指针式表盘的进一步识别工作,将数据标签按照4:1的比例随机划分成训练集和测试集,且使每种不同的水表图片,在训练集和测试集中都有存在。具体标注可以分“数字0”、“数字1”、“数字2”……“数字9”共11类(包含背景),亦或更加细分为“数字0”、“数字0.5”、“数字1.0”……“数字9”、“数字9.5”共21类。三构建基于深度学习网络的检测模型如图2所示,检测模型基于RFCN网络,根据水表中的指针式表盘特性,进行结构改进,整体结构分为基于残差网络的特征提取模块、用于获取感兴趣区域的RPN模块、用于获取分数地图的分类模块及位置敏感的ROIPooling模块。构建残差网络的特征提取模块:本专利技术采用的基于残差网络的特征提取模块,考虑到残差网络计算量小,易于优化,收敛速度块等优势。采用ResNet101的conv1、conv2_x、conv3_x、conv4_x四部分,仅需进行一次特征提取工作,即可共享给后续的改进RPN模块和改进分类模块,大大节省了因多次特征提取的运算时间,其主要结构如表1所示。表1构建用于获取感兴趣区域(ROI)的改进RPN模块:改进的RPN模块分为对区域与anchor的偏移量计算,和对区域隶属于不同类别概率的计算两部分。Anchor是提前设定好的不同尺度的矩形框,本模块前一部分是将区域的偏移量和anchor计算出对应区域。把所有区域和对应的概率值进行排序,进行非极大值抑制算法(NMS)筛选出感兴趣区域。同时,为了解决因网络过深导致的检测准确率降低问题,在计算每个区域的偏移量前增加一个bottleneck结构的残差块,主要结构和关键参数如图3所示。优选地,在rpn_branch2a和rpn_branch2b两个卷积层后增加对应的BatchNorm层、Scale层、Relu层,在rpn_branch2c卷积层后增加对应的BatchNorm层、Scale层。改进的rpn层主要结构有,rpn_conv/3×3、rpn_branch本文档来自技高网...

【技术保护点】
1.一种基于深度学习的指针式水表读数检测方法,其特征在于,包括如下步骤:S1采集含有指针式表盘的水表图片,对图像进行预处理;S2对预处理后的图像指针表盘进行标注,制作数据标签,将数据标签随机划分成训练集和测试集;S3构建基于深度学习网络的检测模型;S4使用训练集训练检测模型,得到适用于指针水表读数的检测模型,并将测试集输入训练好的检测模型中,测试检测模型的性能。

【技术特征摘要】
1.一种基于深度学习的指针式水表读数检测方法,其特征在于,包括如下步骤:S1采集含有指针式表盘的水表图片,对图像进行预处理;S2对预处理后的图像指针表盘进行标注,制作数据标签,将数据标签随机划分成训练集和测试集;S3构建基于深度学习网络的检测模型;S4使用训练集训练检测模型,得到适用于指针水表读数的检测模型,并将测试集输入训练好的检测模型中,测试检测模型的性能。2.根据权利要求1所述的指针式水表读数检测方法,其特征在于,所述S3中基于深度学习网络的检测模型包括基于残差网络的特征提取模块、用于获取感兴趣区域的RPN模块、用于获取分数地图的分类模块及位置敏感的ROIPooling模块。3.根据权利要求2所述的指针式水表读数检测方法,其特征在于,所述RPN模块还包括bottleneck结构的残差块,RPN模块通过对预处理后图片每个区域与anchor的偏移量计算及区域隶属于不同类别概率计算,然后根据偏移量及概率筛选出感兴趣区域。4.根据权利要求2所述的指针式水表读数检测方法,其特征在于,...

【专利技术属性】
技术研发人员:高学赵经纬
申请(专利权)人:华南理工大学
类型:发明
国别省市:广东,44

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

1