[发明专利]一种基于机器视觉的水位标尺定位及水位计算方法

说明书

本发明涉及水文测站监测技术领域，具体为一种基于机器视觉的水位标尺定位及水位计算方法，包括摄像头和边缘计算设备，所述摄像头主要提供实时视频流，所述水位标尺定位及水位计算方法包括以下步骤：(1)训练模型；(2)水尺检测；(3)对步骤1裁剪后的水尺图片进行预处理；(4)第一次提取水位；(5)二次提取水位；(6)对裁剪后的图片进行水尺数字识别，对数字进行校正和排序；(7)水尺精度计算；(8)计算实际水位，用步骤6计算的数字减去步骤7计算的距离就为当前的实际水位。该发明具有成本低廉，只需添加一台边缘计算设备，可管控多路摄像头，无需额外添加其他设备、实时性强，实时读取摄像头视频流，准确性高。

技术领域

本发明涉及水文测站监测技术领域，具体为一种基于机器视觉的水位标尺定位及水位计算方法。

背景技术

水位是河道湖泊的基本水文要素之一，准确、可靠、及时的水位检测是对水文站的最基本要求，目前水位检测的方式主要为人工调取摄像头读取水尺水位信息或者通过添加水位读取设备的方式读取水尺水位信息，现有的自动水位计主要包括遥测浮子式水位计、雷达式水位计、压力式水位计、超声波式水位计、电子水尺等几种，人工调取摄像头观测水尺水位，自动化程度低、耗费人力。遥测浮子式水位计需要建设水井，造价成本过高且需要保持水井内水源清晰且要防止水井淤积，雷达式水位计的精度受水面漂浮物的影响，压力式水位计受泥沙影响精度，压阻式有时飘、温飘，要定时率定。超声波式水位计有温飘，水面漂浮物影响精度，要定时校正，电子水尺野外环境影响正常工作，适用于浅水检测。

目前已存在的利用机器视觉的读取水位的算法适用性较差，遇到特殊情况计算误差较大。

发明内容

本发明的目的是提供一种基于机器视觉的水位标尺定位及水位计算方法。

本发明的上述发明目的是通过以下技术方案得以实现的：

一种基于机器视觉的水位标尺定位及水位计算方法，包括摄像头和边缘计算设备，所述摄像头主要提供实时视频流，以供实时提取水尺图像，边缘计算设备主要提供水尺及水尺上数字的检测和定位，和水尺水位的算法，根据摄像头的IP地址，生成摄像头实时视频流的RTSP地址，接入到边缘设备中，边缘设备自动读取摄像头流地址，截取视频帧，进行算法计算操作，所述水位标尺定位及水位计算方法包括以下步骤：

(1)训练模型，采用YOLO算法训练水尺模型，训练出的模型留作下方水尺检测算法使用，裁剪出原始数据集中的水尺，采用YOLO算法训练水尺数字刻度模型，训练出的模型留作下方水尺数字刻度检测算法使用；

(2)水尺检测，检测水尺，设定YOLO算法中的置信度，把检测出错误的水尺的做筛选，以确定检测到的水尺为正确的水尺，确定水尺的边框裁剪出水尺图片，留作下一步使用；

(3)对步骤1裁剪后的水尺图片进行预处理，得到图像边缘轮廓，根据直线计算水尺角度，使其处于正角度，该图像即为垂直的水尺图像，对裁剪后的图像留作下一步使用；

(4)第一次提取水位，对检测到的水尺预处理，计算出当前水尺图片的主颜色，对步骤3得到的结果图片进行主颜色提取，对提取后的图像进行二值化处理，开运算、闭运算操作，对图像进行边缘检测，获取图像边缘，根据轮廓边缘确定水尺图片最左边、最右边、下方需要裁剪的位置，该位置就为第一次提取水位位置，根据水位位置裁剪图片，留作下一步使用；

(5)二次提取水位，步骤4可以获取到一般情况下的正常水位，当遇到特殊情况下时就需要二次提取水位，二次提取水位不影响一次提取水位的结果；

(6)对裁剪后的图片进行水尺数字识别，对数字进行校正和排序；

(7)水尺精度计算；

(8)计算实际水位，用步骤6计算的数字减去步骤7计算的距离就为当前的实际水位。