基于轻量特征融合网络的粒子图像测速研究

Particle Image Velocimetry Estimation Method Based on Lightweight Feature Fusion Network

  • 摘要: 粒子图像测速技术是一种对流体速度场进行近瞬态、非侵入式测量的重要实验手段。近年来,随着深度学习技术的快速发展,利用深度卷积网络可有效提升粒子图像测速的估计精度,但相关方法中存在网络参数量大等问题。为了有效减少粒子图像测速深度网络的参数量,本文提出了一种基于轻量特征融合网络的粒子图像测速方法。首先,基于深度可分离卷积与增强型ShuffleNet模块,设计轻量化骨干网络,对流场特征进行多尺度编码;其次,基于深度可分离卷积残差结构,对流场的多尺度特征进行层级融合与解码;最后,利用所融合的流场特征,对流场速度进行推理预测。利用公开数据集对所提出的方法进行了定量验证,实验结果表明,在不同类型的多个流场中,本文方法估计精度优于传统PIV方法,较H-S光流方法精度平均提升48%;同时优于多种深度学习方法,较LiteFlowNet-en方法精度平均提升42%。所提出方法的模型参数量仅为0.19M,为LiteFlowNet-en方法参数量的3.4%,PIV-RAFT-2P方法参数量的5.1%,大大提高了PIV估计的计算效率。

     

    Abstract: Particle Image Velocimetry (PIV) technology is an important experimental method for near-instantaneous, non-intrusive measurement of fluid velocity fields. In recent years, with the rapid development of deep learning, the use of convolutional neural networks can effectively improve the estimation accuracy of PIV. However, these methods often suffer from the issue of large network parameter sizes. To effectively reduce the parameter size of deep networks for PIV, a lightweight feature fusion network-based method for PIV is proposed in this paper. Firstly, a lightweight backbone network is designed based on depthwise separable convolutions and enhanced ShuffleNet modules to perform multi-scale feature encoding on particle velocity images. Secondly, a feature fusion and decoding module is designed to hierarchically fuse and decode the extracted fluid field features. Finally, the flow field velocity is inferred and predicted based on the fused fluid field features. The proposed method is quantitatively validated using publicly available datasets. Experimental results demonstrate that the estimation accuracy of the proposed method outperforms traditional PIV methods across various types of flow fields, achieving an average accuracy improvement of 48% compared to the H-S optical flow method. It also surpasses several deep learning-based methods, with an average accuracy improvement of 42% over the LiteFlowNet-en method. Furthermore, the proposed method's model parameter size is only 0.19M, which is 3.4% of LiteFlowNet-en's parameter size and 5.1% of PIV-RAFT-2P's parameter size, significantly enhancing the computational efficiency of PIV estimation.

     

/

返回文章
返回