Accurately obtaining the distribution of the open-channel velocity field in hydraulic engineering is extremely important, which is helpful for better calculation of open-channel flow and analysis of open-channel water flow characteristics. In recent years, machine learning has been used for open-channel velocity field prediction. However, effective training of data-driven models in machine learning heavily depends on the diversity and quantity of data. In this paper, a CFD-based pre-training neural network model (CFD–PNN) is proposed for accurate open-channel velocity field prediction, allowing the adaption to the task with small sample data. Also, a cross-sectional velocity field prediction method combining the computational fluid dynamics (CFD) and machine learning is established. By comparing CFD–PNN with six other neural network algorithm models and the CFD model, the results show that, in the case of small sample data, the CFD–PNN model can predict a more reasonable open-channel velocity field with higher prediction accuracy than other models. The average error of the velocity calculation for the trapezoidal open-channel cross-section is about 3.62%. Compared with other models, the accuracy is improved by 0.3–2.8%.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.