Back to Search Start Over

Image inpainting network based on multi‐level attention mechanism.

Authors :
Xiang, Hongyue
Min, Weidong
Wei, Zitai
Zhu, Meng
Liu, Mengxue
Deng, Ziyang
Source :
IET Image Processing (Wiley-Blackwell). 2/7/2024, Vol. 18 Issue 2, p428-438. 11p.
Publication Year :
2024

Abstract

Image inpainting networks based on deep learning techniques have been widely used in many important fields. However, most inpainting networks fail to generate desirable repaired images. This may be due to their failure to extract effective features and accurately assign high weights to the undamaged regions. To alleviate these problems, an image inpainting network based on gated convolution and multi‐level attention mechanism (IIN‐GCMAM) is proposed in this paper. This network follows encoder–decoder architecture, consisting of the gated convolution encoder (GC‐encoder) and the multi‐level attention mechanism decoder (MAM‐decoder). The GC‐encoder weighs the extracted features with gated convolutions, which reduces the interference caused by the damaged regions. The multi‐level attention mechanism employed in the MAM‐decoder uses multi‐scale feature maps spatially and channel‐wise to improve the consistency in global structure and the fineness of repaired results. Extensive experiments are conducted on the common datasets, Paris StreetView and CelebA. Experimental results indicate that the proposed IIN‐GCMAM can achieve a good performance on the common evaluation metrics and visual effects. It can achieve 0.0408, 0.720, and 22.27 in MAE, SSIM, and PSNR at the mask ratio of 50%–60%, respectively. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
17519659
Volume :
18
Issue :
2
Database :
Academic Search Index
Journal :
IET Image Processing (Wiley-Blackwell)
Publication Type :
Academic Journal
Accession number :
175229216
Full Text :
https://doi.org/10.1049/ipr2.12958