流速分布的预测方法、装置、存储介质及计算机设备

本申请实施例公开了一种流速分布的预测方法、装置、存储介质及计算机设备，涉及计算机视觉领域。本申请先用第一样本集训练得到第一模型。接着对样本集视频帧随机旋转，再划分成两个分组。针对第一分组视频帧，确定河道区域后依掩码块参数设正方形掩码块；对第二分组，同样确定河道区域，按掩码条参数设数量为1、长度匹配河道区域长度、宽度关联河道宽度的掩码条。将掩码处理后的两分组合并为第二样本集，调整第一模型参数得第二模型。最后用训练好的第二模型处理输入的河流监测视频，输出流速分布图，可以提高预测结果的准确性。