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

SeD: Semantic-Aware Discriminator for Image Super-Resolution

About

Generative Adversarial Networks (GANs) have been widely used to recover vivid textures in image super-resolution (SR) tasks. In particular, one discriminator is utilized to enable the SR network to learn the distribution of real-world high-quality images in an adversarial training manner. However, the distribution learning is overly coarse-grained, which is susceptible to virtual textures and causes counter-intuitive generation results. To mitigate this, we propose the simple and effective Semantic-aware Discriminator (denoted as SeD), which encourages the SR network to learn the fine-grained distributions by introducing the semantics of images as a condition. Concretely, we aim to excavate the semantics of images from a well-trained semantic extractor. Under different semantics, the discriminator is able to distinguish the real-fake images individually and adaptively, which guides the SR network to learn the more fine-grained semantic-aware textures. To obtain accurate and abundant semantics, we take full advantage of recently popular pretrained vision models (PVMs) with extensive datasets, and then incorporate its semantic features into the discriminator through a well-designed spatial cross-attention module. In this way, our proposed semantic-aware discriminator empowered the SR network to produce more photo-realistic and pleasing images. Extensive experiments on two typical tasks, i.e., SR and Real SR have demonstrated the effectiveness of our proposed methods.

Bingchen Li, Xin Li, Hanxin Zhu, Yeying Jin, Ruoyu Feng, Zhizheng Zhang, Zhibo Chen• 2024

Related benchmarks

TaskDatasetResultRank
Image Super-resolutionUrban100
PSNR26.46
221
Image Super-resolutionSet14 classic (test)
PSNR27.94
52
Super-ResolutionDIV2K (val)
PSNR29.85
44
Image Super-resolutionManga109
LPIPS0.044
38
Image Super-resolutionSet5 classic (test)
LPIPS0.061
14
Real-world Image Super-ResolutionDPED
NIQE4.49
6
Real-world Image Super-ResolutionOST300
NIQE2.73
6
Real-world Image Super-ResolutionRealSRSet
NIQE5.15
6
Classical Image Super-ResolutionLSDIR
LPIPS0.116
3
Classical Image Super-ResolutionHQ-50K
LPIPS0.157
3
Showing 10 of 10 rows

Other info

Code

Follow for update