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

CRAFT:用于鲁棒光流的交叉注意流Transformer

CRAFT: Cross-Attentional Flow Transformer for Robust Optical Flow

Institute of High Performance Computing, A*STAR(A*STAR高性能计算研究所);Institute for Infocomm Research, A*STAR(A*STAR资讯通信研究院)

Xiuchao Sui, Shaohua Li, Xue Geng, Yan Wu, Xinxing Xu, Yong Liu, Rick Goh, Hongyuan Zhu

摘要:

光流估计旨在通过识别两幅图像之间的对应像素来找到2D运动场。尽管基于深度学习的光流方法取得了巨大进步,但准确估计具有运动模糊的大位移仍然是一个挑战。这主要是因为相关体积,像素匹配的基础,被计算为两个图像的卷积特征的点积。卷积特征的局部性使得计算的相关性容易受到各种噪声的影响。在具有运动模糊的大位移上,噪声相关性可能导致估计流量的严重误差。为了克服这一挑战,我们提出了一个新的架构“交叉注意流Transformer”(CRAFT),旨在振兴相关体积计算。在CRAFT中,语义平滑Transformer层转换一个帧的特征,使它们更具全局性和语义稳定性。此外,点积相关性被替换为Transformer跨帧注意。该层通过查询和关键字投影过滤掉特征噪声,并计算更准确的相关性。在Sintel(Final)和KITTI(前台)基准测试中,CRAFT实现了最先进的性能。此外,为了测试不同模型对大运动的鲁棒性,我们设计了一种图像移位攻击,该攻击将输入图像移位以生成大的人工运动。在这种攻击下,CRAFT比两种代表性的方法RAFT和GMA表现得更鲁棒。CRAFT的代码可在https://github.com/askerlee/craft上获得。

英文摘要:

Optical flow estimation aims to find the 2D motion field by identifying corresponding pixels between two images. Despite the tremendous progress of deep learning-based optical flow methods, it remains a challenge to accurately estimate large displacements with motion blur. This is mainly because the correlation volume, the basis of pixel matching, is computed as the dot product of the convolutional features of the two images. The locality of convolutional features makes the computed correlations susceptible to various noises. On large displacements with motion blur, noisy correlations could cause severe errors in the estimated flow. To overcome this challenge, we propose a new architecture "CRoss-Attentional Flow Transformer" (CRAFT), aiming to revitalize the correlation volume computation. In CRAFT, a Semantic Smoothing Transformer layer transforms the features of one frame, making them more global and semantically stable. In addition, the dot-product correlations are replaced with transformer Cross-Frame Attention. This layer filters out feature noises through the Query and Key projections, and computes more accurate correlations. On Sintel (Final) and KITTI (foreground) benchmarks, CRAFT has achieved new state-of-the-art performance. Moreover, to test the robustness of different models on large motions, we designed an image shifting attack that shifts input images to generate large artificial motions. Under this attack, CRAFT performs much more robustly than two representative methods, RAFT and GMA. The code of CRAFT is is available at https://github.com/askerlee/craft.

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

↑