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

Bridging Global Context Interactions for High-Fidelity Image Completion

About

Bridging global context interactions correctly is important for high-fidelity image completion with large masks. Previous methods attempting this via deep or large receptive field (RF) convolutions cannot escape from the dominance of nearby interactions, which may be inferior. In this paper, we propose to treat image completion as a directionless sequence-to-sequence prediction task, and deploy a transformer to directly capture long-range dependence in the encoder. Crucially, we employ a restrictive CNN with small and non-overlapping RF for weighted token representation, which allows the transformer to explicitly model the long-range visible context relations with equal importance in all layers, without implicitly confounding neighboring tokens when larger RFs are used. To improve appearance consistency between visible and generated regions, a novel attention-aware layer (AAL) is introduced to better exploit distantly related high-frequency features. Overall, extensive experiments demonstrate superior performance compared to state-of-the-art methods on several datasets.

Chuanxia Zheng, Tat-Jen Cham, Jianfei Cai, Dinh Phung• 2021

Related benchmarks

TaskDatasetResultRank
Image InpaintingPlaces2 (test)--
68
Camouflaged Image SynthesisLAKE-RED Camouflaged Objects
KL_BF0.3399
28
Camouflaged Image SynthesisLAKE-RED Salient Objects
KL_BF0.6743
14
Camouflaged Image SynthesisLAKE-RED (Overall)
KL_BF0.5909
14
Image CompletionPlaces2 20-30% mask ratio
PSNR25.1
7
Image CompletionPlaces2 30-40% mask ratio
PSNR22.89
7
Image CompletionPlaces2 40-50% mask ratio
PSNR21.22
7
Confetti removalRealistic scenes Indoor Apple iPhone 12
PSNR20.91
4
Confetti removalRealistic scenes Apple iPhone 12 (outdoor)
PSNR21.66
4
Showing 9 of 9 rows

Other info

Code

Follow for update