A Simple Plugin for Transforming Images to Arbitrary Scales

Qinye Zhou* 1
Ziyi Li* 1
Weidi Xie1, 2
Xiaoyun Zhang1
Yanfeng Wang1, 2
Ya Zhang1, 2

1CMIC, Shanghai Jiao Tong University
2Shanghai AI Laboratory

BMVC 2022



Code

Paper

Bibtex



alt text


Abstract

Existing models on super-resolution often specialized for one scale, fundamentally limiting their use in practical scenarios. In this paper, we aim to develop a general plugin that can be inserted into existing super-resolution models, conveniently augmenting their ability towards Arbitrary Resolution Image Scaling, thus termed ARIS. We make the following contributions: (1) we propose a transformer-based plugin module, which uses spatial coordinates as query, iteratively attend the low-resolution image feature through cross-attention, and output visual feature for the queried spatial location, resembling an implicit representation for images; (2) we introduce a novel self-supervised training scheme, that exploits consistency constraints to effectively augment the model's ability for upsampling images towards unseen scales, i.e. ground-truth high-resolution images are not available; (3) without loss of generality, we inject the proposed ARIS plugin module into several existing models, namely, IPT, SwinIR, and HAT, showing that the resulting models can not only maintain their original performance on fixed scale factor but also extrapolate to unseen scales, substantially outperforming existing any-scale super-resolution models on standard benchmarks, e.g. Urban100, DIV2K, etc.



Training stragety

alt text


Results

Arbitrary-scale super-resolution results on Set5, Set14, Urban100, B100 and DIV2K.

Ablation study result on training stragety

Ablation study result on architecture

Qualitative Results

Webpage template modified from Richard Zhang.