一种基于卷积神经网络的河流表面流速测量方法及系统技术方案

本发明专利技术提供一种基于卷积神经网络的河流表面流速测量方法及系统，属于水文测量技术领域，包括：获取河流表面图像数据集，对河流表面图像数据集进行预处理，得到预处理图像数据集；利用

【技术实现步骤摘要】
一种基于卷积神经网络的河流表面流速测量方法及系统


[0001]本专利技术涉及水文测量
，尤其涉及一种基于卷积神经网络的河流表面流速测量方法及系统
。

技术介绍

[0002]近年来，随着在智慧城市
、
智慧水利的普遍兴起，计算机视觉技术结合水文学和流体力学学科在流体运动研究方面不断进行发展
。
河流流速是水文信息获取的关键，掌握实时
、
有效且全面的水文信息可以使人们能够及时有效应对频繁的洪涝灾害，减少其会带来的人力
、
物力和财力损失
。
[0003]传统的流速测量依赖于各种仪器，有通过水流带动旋桨转动，依据旋桨转速计算的传统流速仪测量法，有通过声学
、
光学多普勒效应进行测量的流速法，还有基于视觉图像的图像法测速，更加适用于极端环境测流条件
。
但上述方法受限于河流环境等因素影响，测量精度得不到保障
。
[0004]在此基础上，采用深度学习的测量方法得到了广泛应用，该方法为无接触测量方法，通过收集大量的河流表面流速数据样本，建立网络模型进行训练，通过构建深层神经网络模型，从大量数据中学习特征表示，实现对光流的估计，在一定程度上克服了传统方法中对复杂场景
、
光照变化和噪声等问题的敏感性，但仍存在受限于样本数据质量，模型泛化能力较差等问题，无法适应复杂多变的河流环境
。

技术实现思路

[0005]本专利技术提供一种基于卷积神经网络的河流表面流速测量方法及系统，用以解决现有技术中采用深度学习时，构建的神经网络模型受限于样本数据质量，模型泛化能力较差，缺乏普适性的缺陷
。
[0006]第一方面，本专利技术提供一种基于卷积神经网络的河流表面流速测量方法，包括：
[0007]获取河流表面图像数据集，对所述河流表面图像数据集进行预处理，得到预处理图像数据集；
[0008]利用
DeepFlow
算法计算所述预处理图像数据集获得光流矢量；
[0009]采用
mobilenet v2
网络构建初始河流测速模型，通过预设损失函数和所述光流矢量对所述初始河流测速模型进行训练迭代，得到河流测速模型；
[0010]将待测量河流视频流输入所述河流测速模型，得到预测河流表面流速
。
[0011]在一个实施例中，获取河流表面图像数据集，对所述河流表面图像数据集进行预处理，得到预处理图像数据集，包括：
[0012]采集河面原始视频数据，采用预设帧格式截取多帧图像数据；
[0013]对所述多帧图像数据进行去噪，将去噪后的多帧图像按照预设比例划分为训练集和测试集；
[0014]获取真实流速测量数据，以所述真实流速测量数据作为标签，基于所述训练集和
所述测试集形成所述预处理图像数据集
。
[0015]在一个实施例中，利用
DeepFlow
算法计算所述预处理图像数据集获得光流矢量，包括：
[0016]利用卷积神经网络从所述预处理图像数据集中提取图像特征；
[0017]获取所述图像特征中的相邻帧特征点，确定所述相邻帧特征点的位移匹配信息；
[0018]基于光流估计能量最小化算法，计算所述位移匹配信息得到光流场，提取所述光流场中的光流矢量
。
[0019]在一个实施例中，基于光流估计能量最小化算法，计算所述位移匹配信息得到光流场，提取所述光流场中的光流矢量，包括：
[0020]确定数据项
、
平滑项和匹配项，基于所述数据项
、
所述平滑项
、
平滑项平衡因子
、
所述匹配项和匹配项平衡因子构建能量函数，其中所述数据项通过比较两帧图像之间亮度差异确定光流估计准确性，所述平滑项通过比较相邻像素之间光流差异确定光流场平滑性；
[0021]基于深度学习对所述能量函数进行迭代优化，训练所述平滑项平衡因子和所述匹配项平衡因子，得到最优能量函数；
[0022]由所述最优能量函数确定所述光流场，提取所述光流场中的速度向量和方向向量
。
[0023]在一个实施例中，采用
mobilenet v2
网络构建初始河流测速模型，通过预设损失函数和所述光流场对所述初始河流测速模型进行训练迭代，得到河流测速模型，包括：
[0024]采用均方误差
L2损失函数和河面流速值为主损失函数，采用均方误差
L2损失函数
、
光流矢量速度和光流矢量角度为辅助损失函数，构建所述预设损失函数；
[0025]计算所述预设损失函数对所述初始河流测速模型的梯度信息，在所述初始河流测速模型的输出层向输入层的前向传播中，逐层更新参数，采用链式法计算所述梯度信息得到误差，将所述误差逐层进行反向传播；
[0026]基于所述梯度信息更新最优能量函数中的平滑项平衡因子和匹配项平衡因子，使所述预设损失函数逐渐减小；
[0027]重复执行前向传播
、
损失计算
、
更新参数和反向传播的步骤，直至满足预设迭代次数；
[0028]采用训练集进行模型训练，采用测试集进行模型性能评估，得到所述河流测速模型
。
[0029]在一个实施例中，采用均方误差
L2损失函数和河面流速值为主损失函数，采用均方误差
L2损失函数
、
光流矢量速度和光流矢量角度为辅助损失函数，构建所述预设损失函数，包括：
[0030]确定光流矢量总数；
[0031]采用所述均方误差
L2损失函数计算河面流速真实值
、
河面流速预测值以及所述光流矢量总数，得到所述主损失函数；
[0032]获取任一当前时刻光流矢量速度和任一当前时刻的上一时刻光流矢量速度，采用所述均方误差
L2损失函数计算所述任一当前时刻光流矢量速度
、
所述上一时刻光流矢量速度和所述光流矢量总数，得到光流矢量速度损失函数；
[0033]采用所述均方误差
L2损失函数计算光流矢量角度真实值
、
光流矢量角度预测值以
及所述光流矢量总数，得到光流矢量角度损失函数；
[0034]根据所述主损失函数
、
所述光流矢量速度损失函数
、
光流矢量速度平衡因子
、
所述光流矢量角度损失函数和光流矢量角度平衡因子，得到所述预设损失函数
。
[0035]在一个实施例中，采用
mobilenet v2
网络构建初始河流测速模型，包括：
[0036]确定所述
mobilenet v2
网络包括1个卷积层
、17
个倒残差结构层和2个点卷积层，所述
17
个倒残差结构层本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.
一种基于卷积神经网络的河流表面流速测量方法，其特征在于，包括：获取河流表面图像数据集，对所述河流表面图像数据集进行预处理，得到预处理图像数据集；利用
DeepFlow
算法计算所述预处理图像数据集获得光流矢量；采用
mobilenet v2
网络构建初始河流测速模型，通过预设损失函数和所述光流矢量对所述初始河流测速模型进行训练迭代，得到河流测速模型；将待测量河流视频流输入所述河流测速模型，得到预测河流表面流速
。2.
根据权利要求1所述的基于卷积神经网络的河流表面流速测量方法，其特征在于，获取河流表面图像数据集，对所述河流表面图像数据集进行预处理，得到预处理图像数据集，包括：采集河面原始视频数据，采用预设帧格式截取多帧图像数据；对所述多帧图像数据进行去噪，将去噪后的多帧图像按照预设比例划分为训练集和测试集；获取真实流速测量数据，以所述真实流速测量数据作为标签，基于所述训练集和所述测试集形成所述预处理图像数据集
。3.
根据权利要求1所述的基于卷积神经网络的河流表面流速测量方法，其特征在于，利用
DeepFlow
算法计算所述预处理图像数据集获得光流矢量，包括：利用卷积神经网络从所述预处理图像数据集中提取图像特征；获取所述图像特征中的相邻帧特征点，确定所述相邻帧特征点的位移匹配信息；基于光流估计能量最小化算法，计算所述位移匹配信息得到光流场，提取所述光流场中的光流矢量
。4.
根据权利要求3所述的基于卷积神经网络的河流表面流速测量方法，其特征在于，基于光流估计能量最小化算法，计算所述位移匹配信息得到光流场，提取所述光流场中的光流矢量，包括：确定数据项
、
平滑项和匹配项，基于所述数据项
、
所述平滑项
、
平滑项平衡因子
、
所述匹配项和匹配项平衡因子构建能量函数，其中所述数据项通过比较两帧图像之间亮度差异确定光流估计准确性，所述平滑项通过比较相邻像素之间光流差异确定光流场平滑性；基于深度学习对所述能量函数进行迭代优化，训练所述平滑项平衡因子和所述匹配项平衡因子，得到最优能量函数；由所述最优能量函数确定所述光流场，提取所述光流场中的速度向量和方向向量
。5.
根据权利要求1所述的基于卷积神经网络的河流表面流速测量方法，其特征在于，采用
mobilenet v2
网络构建初始河流测速模型，通过预设损失函数和所述光流矢量对所述初始河流测速模型进行训练迭代，得到河流测速模型，包括：采用均方误差
L2损失函数和河面流速值为主损失函数，采用均方误差
L2损失函数
、
光流矢量速度和光流矢量角度为辅助损失函数，构建所述预设损失函数；计算所述预设损失函数对所述初始河流测速模型的梯度信息，在所述初始河流测速模型的输出层向输入层的前向传播中，逐层更新参数，采用链式法计算所述梯度信息得到误差，将所述误差逐层进行反向传播；基于所述梯度信息更新最优能量函数中的平滑项平衡因子和匹配项平衡因子，使所述
预设损失函数逐渐减小；重复执行前向传播
、
损失计算
、
更新参数和反向传播的步骤，直至满足预设迭代次数；采用训练集进行模型训练，采用测试集进行模型性能评估，得到所述河流测速模型
。6.
根据权利要求5所述的基于卷积神经网络的河流表面流速测量方法，其特征在于，采用均方误差
L2损失函数和河面流速值为主损失函数，采用均方误差
L2损失函数
、
光流...

【专利技术属性】
技术研发人员：刘炳义，刘维高，郑晓燕，张雨，毕程程，
申请(专利权)人：武汉大水云科技有限公司，
类型：发明
国别省市：