论文标题

无监督的粒子图像速度学学习

Unsupervised Learning of Particle Image Velocimetry

论文作者

Zhang, Mingrui, Piggott, Matthew D.

论文摘要

粒子图像速度法(PIV)是一个经典的流动估计问题,被广泛考虑和使用,尤其是作为实验流体动力学和环境流遥感的诊断工具。最近,基于深度学习的方法的发展激发了解决PIV问题的新方法。这些基于学习的基于学习的方法是由大量数据和地面真相培训信息驱动的。但是,在大规模的现实情况下,很难收集可靠的地面真相数据。尽管合成数据集可以用作替代方案,但训练设置和现实情况之间的差距限制了适用性。我们在这里介绍我们认为是第一批基于学习的方法来解决PIV问题的方法。所提出的方法灵感来自经典的光流方法。我们不使用地面真相数据,而是利用两个连续的图像框架之间的光度损失,双向流量估计的一致性损失和空间平滑度损失来构建总无监督损失函数。该方法显示出流体流量估计的巨大潜力和优势。此处介绍的结果表明,我们的方法与经典PIV方法以及对广泛PIV数据集的基于监督的学习方法相比,输出竞争性结果,甚至在某些困难的流动案例中甚至优于这些现有方法。可以在https://github.com/erizmr/unliteflownet-piv上获得代码和训练的模型。

Particle Image Velocimetry (PIV) is a classical flow estimation problem which is widely considered and utilised, especially as a diagnostic tool in experimental fluid dynamics and the remote sensing of environmental flows. Recently, the development of deep learning based methods has inspired new approaches to tackle the PIV problem. These supervised learning based methods are driven by large volumes of data with ground truth training information. However, it is difficult to collect reliable ground truth data in large-scale, real-world scenarios. Although synthetic datasets can be used as alternatives, the gap between the training set-ups and real-world scenarios limits applicability. We present here what we believe to be the first work which takes an unsupervised learning based approach to tackle PIV problems. The proposed approach is inspired by classic optical flow methods. Instead of using ground truth data, we make use of photometric loss between two consecutive image frames, consistency loss in bidirectional flow estimates and spatial smoothness loss to construct the total unsupervised loss function. The approach shows significant potential and advantages for fluid flow estimation. Results presented here demonstrate that our method outputs competitive results compared with classical PIV methods as well as supervised learning based methods for a broad PIV dataset, and even outperforms these existing approaches in some difficult flow cases. Codes and trained models are available at https://github.com/erizmr/UnLiteFlowNet-PIV.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源