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

Image Completion with Heterogeneously Filtered Spectral Hints

About

Image completion with large-scale free-form missing regions is one of the most challenging tasks for the computer vision community. While researchers pursue better solutions, drawbacks such as pattern unawareness, blurry textures, and structure distortion remain noticeable, and thus leave space for improvement. To overcome these challenges, we propose a new StyleGAN-based image completion network, Spectral Hint GAN (SH-GAN), inside which a carefully designed spectral processing module, Spectral Hint Unit, is introduced. We also propose two novel 2D spectral processing strategies, Heterogeneous Filtering and Gaussian Split that well-fit modern deep learning models and may further be extended to other tasks. From our inclusive experiments, we demonstrate that our model can reach FID scores of 3.4134 and 7.0277 on the benchmark datasets FFHQ and Places2, and therefore outperforms prior works and reaches a new state-of-the-art. We also prove the effectiveness of our design via ablation studies, from which one may notice that the aforementioned challenges, i.e. pattern unawareness, blurry textures, and structure distortion, can be noticeably resolved. Our code will be open-sourced at: https://github.com/SHI-Labs/SH-GAN.

Xingqian Xu, Shant Navasardyan, Vahram Tadevosyan, Andranik Sargsyan, Yadong Mu, Humphrey Shi• 2022

Related benchmarks

TaskDatasetResultRank
Image InpaintingFFHQ 256x256 (val)
FID4.3459
22
InpaintingPlaces wide masks 512 x 512
FID3.9755
20
InpaintingPlaces narrow masks 512 x 512
FID0.9307
20
Image InpaintingPlaces2 512x512 (val)
FID7.0277
10
Image InpaintingPlaces2 narrow mask 256x256 (test)
FID1.3084
10
Image InpaintingPlaces2 wide mask 256x256 (test)
FID2.7853
10
Image InpaintingPlaces2 256x256 (val)
FID7.5036
10
Image InpaintingFFHQ 512x512 (val)
FID3.4134
8
Image InpaintingFFHQ narrow mask 512
FID1.2053
8
Image InpaintingFFHQ wide mask 512
FID2.6009
8
Showing 10 of 12 rows

Other info

Code

Follow for update