Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Comparing Correspondences: Video Prediction with Correspondence-wise Losses

About

Image prediction methods often struggle on tasks that require changing the positions of objects, such as video prediction, producing blurry images that average over the many positions that objects might occupy. In this paper, we propose a simple change to existing image similarity metrics that makes them more robust to positional errors: we match the images using optical flow, then measure the visual similarity of corresponding pixels. This change leads to crisper and more perceptually accurate predictions, and does not require modifications to the image prediction network. We apply our method to a variety of video prediction tasks, where it obtains strong performance with simple network architectures, and to the closely related task of video interpolation. Code and results are available at our webpage: https://dangeng.github.io/CorrWiseLosses

Daniel Geng, Max Hamilton, Andrew Owens• 2021

Related benchmarks

TaskDatasetResultRank
Future video predictionCityscapes Next frame
MS-SSIM0.928
13
Future video predictionCityscapes Next 5 frames
MS-SSIM0.839
13
Future video predictionCityscapes Next 10 frames
LPIPS0.217
13
Future video predictionKITTI Next 3 frames
LPIPS0.22
11
Video PredictionCityscapes 9 (test)
MS-SSIM (t+1)92.8
11
Video PredictionCityscapes
MS-SSIM (t+1)92.8
11
Frame InterpolationVimeo-90K septuplet
PSNR35.13
9
Video PredictionKITTI 12 (test)
MS-SSIM (t+1)82
9
Video PredictionKITTI
MS-SSIM (t+1)82
9
Multi-frame video predictionKITTI Next Frame (test)
SSIM0.82
5
Showing 10 of 11 rows

Other info

Code

Follow for update