一种河流表面流速识别数据增强方法及装置

本发明提供一种河流表面流速识别数据增强方法及装置，属于水文测流技术领域，其中，河流表面流速识别数据增强方法包括：获取河流数据集；将流速标签映射到灰度值区间，得到灰度标签图，所述灰度标签图中任意一点的灰度值与流速对应；在各所述河流原图和所述灰度标签图中随机选取感兴趣区域并进行裁剪，得到裁剪图像，将所述裁剪图像作为深度学习模型的训练数据。本发明中的河流表面流速识别数据增强方法，通过将流速标签映射到灰度值，制作灰度标签图像，相较于传统的流速标签图像或者光流图像，可以提高深度学习训练数据的利用效率，同时，采用随机裁剪的方式提供不同的图像感受野来提高模型训练的鲁棒性。