Learnable Gated Temporal Shift Module for Free-form Video Inpainting

Ya-Liang Chang (National Taiwan University), Zhe Yu Liu (National Taiwan University), Kuan-Ying Lee (National Taiwan University), Winston Hsu (National Taiwan University)

Abstract
How to efficiently utilize temporal information to recover videos in a consistent way is the main issue for video inpainting problems. Conventional 2D CNNs have achieved good performance on image inpainting but often lead to temporally inconsistent results where frames will flicker when applied to videos; 3D CNNs can capture temporal information but are computationally intensive and hard to train. In this paper, we present a novel component termed Learnable Gated Temporal Shift Module (LGTSM) for video inpainting models that could effectively tackle arbitrary video masks without additional parameters from 3D convolutions. LGTSM is designed to let 2D convolutions make use of neighboring frames more efficiently, which is crucial for video inpainting. Specifically, in each layer, LGTSM learns to shift some channels to its temporal neighbors so that 2D convolutions could be enhanced to handle temporal information. Meanwhile, a gated convolution is applied to the layer identify the masked areas that are poisoning for conventional convolutions. On the FaceForensics and Free-form Video Inpainting (FVI) dataset, our model achieves state-of-the-art results with simply 33% of parameters and inference time.

DOI
10.5244/C.33.71
https://dx.doi.org/10.5244/C.33.71

Files
Paper (PDF)
Supplementary material (PDF)

BibTeX