arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2111.13680cs.CV
收藏

GMFlow:通过全局匹配学习光流

GMFlow: Learning Optical Flow via Global Matching

Monash University(莫纳什大学);The University of Sydney(悉尼大学);JD Explore Academy(京东探索研究院)

Haofei Xu, Jing Zhang, Jianfei Cai, Hamid Rezatofighi, Dacheng Tao

摘要:

基于学习的光流估计一直以具有用于流回归的卷积的成本体积的流水线为主,其固有地限于局部相关性,因此难以解决大位移的长期挑战。为了缓解这一问题,最先进的框架RAFT逐渐提高其预测质量,通过使用大量的迭代细化,实现显着的性能,但引入线性增加的推理时间。为了实现高精度和高效率,我们通过将光流重新表述为全局匹配问题来完全改造占主导地位的流回归流水线,该全局匹配问题通过直接比较特征相似性来识别对应关系。具体来说,我们提出了一个GMFlow框架,它由三个主要组成部分:一个自定义的Transformer的功能增强,相关性和softmax层的全局特征匹配,和一个自我注意流传播层。我们进一步引入了一个细化步骤,在更高的特征分辨率下重用GMFlow进行残差流预测。我们的新框架在具有挑战性的Sintel基准测试中优于31个改进的RAFT,同时仅使用一个改进并且运行更快,这表明了准确和有效的光流估计的新范例。代码可在https://github.com/haofeixu/gmflow上获得。

英文摘要:

Learning-based optical flow estimation has been dominated with the pipeline of cost volume with convolutions for flow regression, which is inherently limited to local correlations and thus is hard to address the long-standing challenge of large displacements. To alleviate this, the state-of-the-art framework RAFT gradually improves its prediction quality by using a large number of iterative refinements, achieving remarkable performance but introducing linearly increasing inference time. To enable both high accuracy and efficiency, we completely revamp the dominant flow regression pipeline by reformulating optical flow as a global matching problem, which identifies the correspondences by directly comparing feature similarities. Specifically, we propose a GMFlow framework, which consists of three main components: a customized Transformer for feature enhancement, a correlation and softmax layer for global feature matching, and a self-attention layer for flow propagation. We further introduce a refinement step that reuses GMFlow at higher feature resolution for residual flow prediction. Our new framework outperforms 31-refinements RAFT on the challenging Sintel benchmark, while using only one refinement and running faster, suggesting a new paradigm for accurate and efficient optical flow estimation. Code is available at https://github.com/haofeixu/gmflow.

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

↑