RSTT: Real-time Spatial Temporal Transformer for Space-Time Video Super-Resolution

Space-time video super-resolution (STVSR) is the task of interpolating videos with both Low Frame Rate (LFR) and Low Resolution (LR) to produce High-Frame-Rate (HFR) and also High-Resolution (HR) counterparts. The existing methods based on Convolutional Neural Network~(CNN) succeed in achieving visually satisfied results while suffer from slow inference speed due to their heavy architectures. We propose to resolve this issue by using a spatial-temporal transformer that naturally incorporates the spatial and temporal super resolution modules into a single model. Unlike CNN-based methods, we do not explicitly use separated building blocks for temporal interpolations and spatial super-resolutions; instead, we only use a single end-to-end transformer architecture. Specifically, a reusable dictionary is built by encoders based on the input LFR and LR frames, which is then utilized in the decoder part to synthesize the HFR and HR frames. Compared with the state-of-the-art TMNet \cite{xu2021temporal}, our network is $60\%$ smaller (4.5M vs 12.3M parameters) and $80\%$ faster (26.2fps vs 14.3fps on $720\times576$ frames) without sacrificing much performance. The source code is available at https://github.com/llmpass/RSTT.

PDF Abstract CVPR 2022 PDF CVPR 2022 Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Video Frame Interpolation Vid4 - 4x upscaling RSTT-L PSNR 26.43 # 2
SSIM 0.7994 # 2
Parameters 7670000 # 2
Video Frame Interpolation Vid4 - 4x upscaling RSTT-S PSNR 26.29 # 5
SSIM 0.7941 # 5
Parameters 4490000 # 4
Video Frame Interpolation Vid4 - 4x upscaling RSTT-M PSNR 26.37 # 3
SSIM 0.7978 # 3
Parameters 6080000 # 3
Space-time Video Super-resolution Vimeo90K-Fast RSTT-M PSNR 36.78 # 3
SSIM 0.9401 # 3
Space-time Video Super-resolution Vimeo90K-Fast RSTT-S PSNR 36.58 # 4
SSIM 0.9381 # 4
Space-time Video Super-resolution Vimeo90K-Fast RSTT-L PSNR 36.80 # 2
SSIM 0.9403 # 2
Space-time Video Super-resolution Vimeo90K-Medium RSTT-L PSNR 35.66 # 2
SSIM 0.9381 # 3
Space-time Video Super-resolution Vimeo90K-Medium RSTT-M PSNR 35.62 # 3
SSIM 0.9377 # 2
Space-time Video Super-resolution Vimeo90K-Medium RSTT-S PSNR 35.43 # 4
SSIM 0.9358 # 1

Methods