arrow
Return

Deep Generative Model for Image Inpainting With Local Binary Pattern Learning and Spatial Attention

delete2022-01-01
delete35
delete
OA
AI
H
Haiwei Wu
J
Jiantao Zhou *
Y
Yuanman Li
DOI:10.1109/TMM.2021.3111491delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Deep learning (DL) has demonstrated its powerful capabilities in the field of image inpainting. The DL-based image inpainting approaches can produce visually plausible results, but often generate various unpleasant artifacts, especially in the boundary and highly textured regions. To tackle this challenge, in this work, we propose a new end-to-end, two-stage (coarse-to-fine) generative model through combining a local binary pattern (LBP) learning network with an actual inpainting network. Specifically, the first LBP learning network using U-Net architecture is designed to accurately predict the structural information of the missing region, which subsequently guides the second image inpainting network for better filling the missing pixels. Furthermore, an improved spatial attention mechanism is integrated into the image inpainting network, by considering the consistency not only between the known region with the generated one, but also within the generated region itself. Extensive experiments on public datasets including CelebA-HQ, Places and Paris StreetView demonstrate that our model generates better inpainting results than the state-of-the-art competing algorithms, both quantitatively and qualitatively. The source code and trained models are available at https://github.com/HighwayWu/ImageInpainting.
Keywords:
Feature extraction
Generators
Decoding
Task analysis
Semantics
Image edge detection
Correlation
Image inpainting
LBP
spatial attention
deep learning

Journal

IEEE Transactions on Multimedia cover
IEEE Transactions on Multimedia
IF:
9.7
Papers:
4.5K
Citations:
2.4W

Organization

S
shenzhen university
Scholars:
4.5W
Papers: 3.4W
Citations: 72
U
University of Macau
Scholars:
1.1W
Papers: 1.3W
Citations: 2.0W