HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Exploiting Optical Flow Guidance for Transformer-Based Video Inpainting

Kaidong Zhang Jialun Peng Jingjing Fu Dong Liu

Exploiting Optical Flow Guidance for Transformer-Based Video Inpainting

Abstract

Transformers have been widely used for video processing owing to the multi-head self attention (MHSA) mechanism. However, the MHSA mechanism encounters an intrinsic difficulty for video inpainting, since the features associated with the corrupted regions are degraded and incur inaccurate self attention. This problem, termed query degradation, may be mitigated by first completing optical flows and then using the flows to guide the self attention, which was verified in our previous work - flow-guided transformer (FGT). We further exploit the flow guidance and propose FGT++ to pursue more effective and efficient video inpainting. First, we design a lightweight flow completion network by using local aggregation and edge loss. Second, to address the query degradation, we propose a flow guidance feature integration module, which uses the motion discrepancy to enhance the features, together with a flow-guided feature propagation module that warps the features according to the flows. Third, we decouple the transformer along the temporal and spatial dimensions, where flows are used to select the tokens through a temporally deformable MHSA mechanism, and global tokens are combined with the inner-window local tokens through a dual perspective MHSA mechanism. FGT++ is experimentally evaluated to be outperforming the existing video inpainting networks qualitatively and quantitatively.

Code Repositories

hitachinsk/fgt
pytorch
Mentioned in GitHub
hitachinsk/isvi
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
video-inpainting-on-davisFGT++
LPIPS (object): 0.035
LPIPS (square): 0.028
PNSR (object): 35.61
SSIM (object): 0.961
SSIM (square): 0.971
video-inpainting-on-davisFGT++*
LPIPS (object): 0.027
LPIPS (square): 0.022
PNSR (object): 35.9
SSIM (object): 96.8
SSIM (square): 97.6
video-inpainting-on-youtube-vos-1FGT++
LPIPS: 0.025
PSNR: 35.02
PSNR (square): 33.18
SSIM: 97.6
video-inpainting-on-youtube-vos-1FGT++*
LPIPS: 0.022
PSNR: 35.36
PSNR (square): 33.72
SSIM: 97.8

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Exploiting Optical Flow Guidance for Transformer-Based Video Inpainting | Papers | HyperAI