Abstract:Recently, deep learning-based image inpainting methods have made great strides in reconstructing damaged regions. However, these methods often struggle to produce satisfactory results when dealing with missing images with large holes, leading to distortions in the structure and blurring of textures. To address these problems, we combine the advantages of transformers and convolutions to propose an image inpainting method that incorporates edge priors and attention mechanisms. The proposed method aims to improve the results of inpainting large holes in images by enhancing the accuracy of structure restoration and the ability to recover texture details. This method divides the inpainting task into two phases: edge prediction and image inpainting. Specifically, in the edge prediction phase, a transformer architecture is designed to combine axial attention with standard self-attention. This design enhances the extraction capability of global structural features and location awareness. It also balances the complexity of self-attention operations, resulting in accurate prediction of the edge structure in the defective region. In the image inpainting phase, a multi-scale fusion attention module is introduced. This module makes full use of multi-level distant features and enhances local pixel continuity, thereby significantly improving the quality of image inpainting. To evaluate the performance of our method, comparative experiments are conducted on several datasets, including CelebA, Places2, and Facade. Quantitative experiments show that our method outperforms the other mainstream methods. Specifically, it improves Peak Signal-to-Noise Ratio (PSNR) and Structure Similarity Index Measure (SSIM) by 1.141~3.234 db and 0.083~0.235, respectively. Moreover, it reduces Learning Perceptual Image Patch Similarity (LPIPS) and Mean Absolute Error (MAE) by 0.0347~0.1753 and 0.0104~0.0402, respectively. Qualitative experiments reveal that our method excels at reconstructing images with complete structural information and clear texture details. Furthermore, our model exhibits impressive performance in terms of the number of parameters, memory cost, and testing time.

Multi-scale Gated Inpainting Network with Patch-Wise Spacial Attention.

UCTGAN: Diverse Image Inpainting Based on Unsupervised Cross-Space Translation

Deep Generative Network for Image Inpainting with Gradient Semantics and Spatial-Smooth Attention

MSE-Net: Generative Image Inpainting with Multi-Scale Encoder

Image Inpainting with Multi-scale Consistent Features

Image Inpainting Based on Interactive Separation Network and Progressive Reconstruction Algorithm

A Progressive Image Inpainting Algorithm with a Mask Auto-update Branch

Progressive Inpainting Strategy with Partial Convolutions Generative Networks (PPCGN).

High-Resolution Image Inpainting Based On Multi-Scale Neural Network

Multi-scale Generative Adversarial Inpainting Network Based on Cross-Layer Attention Transfer Mechanism.

MFMAM: Image inpainting via multi-scale feature module with attention module

High-Resolution Image Inpainting with Iterative Confidence Feedback and Guided Upsampling

Image Inpainting Using Multi-Scale Feature Joint Attention Model

DeepGIN: Deep Generative Inpainting Network for Extreme Image Inpainting

Multi-Scale Generative Adversarial Networks Image Inpainting Algorithm*

Multi-scale Generative Model for Image Completion

Multi-scale patch-GAN with edge detection for image inpainting

Image Inpainting Technique Incorporating Edge Prior and Attention Mechanism

Structure First Detail Next: Image Inpainting with Pyramid Generator

Semantic Image Inpainting with Multi-Stage Feature Reasoning Generative Adversarial Network

Free-Form Image Inpainting with Separable Gate Encoder-Decoder Network