arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.14487cs.CVcs.LGeess.IV
收藏

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

Unsupervised Learning of Particle Image Velocimetry

Imperial College London(伦敦帝国学院)

Mingrui Zhang, Matthew D. Piggott

摘要:

粒子图像测速技术(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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑