HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Deficiency-Aware Masked Transformer for Video Inpainting

Yongsheng Yu Heng Fan Libo Zhang

Deficiency-Aware Masked Transformer for Video Inpainting

Abstract

Recent video inpainting methods have made remarkable progress by utilizing explicit guidance, such as optical flow, to propagate cross-frame pixels. However, there are cases where cross-frame recurrence of the masked video is not available, resulting in a deficiency. In such situation, instead of borrowing pixels from other frames, the focus of the model shifts towards addressing the inverse problem. In this paper, we introduce a dual-modality-compatible inpainting framework called Deficiency-aware Masked Transformer (DMT), which offers three key advantages. Firstly, we pretrain a image inpainting model DMT_img serve as a prior for distilling the video model DMT_vid, thereby benefiting the hallucination of deficiency cases. Secondly, the self-attention module selectively incorporates spatiotemporal tokens to accelerate inference and remove noise signals. Thirdly, a simple yet effective Receptive Field Contextualizer is integrated into DMT, further improving performance. Extensive experiments conducted on YouTube-VOS and DAVIS datasets demonstrate that DMT_vid significantly outperforms previous solutions. The code and video demonstrations can be found at github.com/yeates/DMT.

Code Repositories

yeates/dmt
Official
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
video-inpainting-on-davisDMT
Ewarp: -
PSNR: 33.82
SSIM: 0.976
VFID: 0.104
video-inpainting-on-youtube-vosDMT
Ewarp: -
PSNR: 34.27
SSIM: 0.9730
VFID: 0.044

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