水面流速检测方法、装置、设备及介质

本公开涉及图像识别技术领域，公开了水面流速检测方法、装置、设备及介质，该方法包括：获取多帧待检测的图像数据；利用目标检测网络对当前帧的图像数据进行特征提取、特征融合以及预测，得到当前帧的图像数据对应的第一目标检测框，对下一帧的图像数据进行特征提取、特征融合以及预测，得到下一帧的图像数据对应的第二目标检测框；利用目标跟踪算法基于第一目标检测框和第二目标检测框，对当前帧下的所有水面示踪物的历史轨迹进行更新，得到当前帧下的所有水面示踪物的目标轨迹；基于目标轨迹得到当前帧的图像数据的水面流速。本公开通过在目标检测网络中融合注意力机制，并利用目标跟踪算法对水面示踪物进行跟踪，提高了水面流速检测精度。