Generative Adversarial Network for Mural Inpainting Based on Multi-level Feature Fusion and Hypergraph Convolution

被引:0
|
作者
Chen Y. [1 ,2 ]
Tao M. [1 ]
Zhao M. [1 ]
机构
[1] School of Elec. and Info. Eng., Lanzhou Jiaotong Univ., Lanzhou
[2] Gansu Provincial Eng. Research Center for Artificial Intelligence and Graphics & Image Processing, Lanzhou
关键词
ConvLSTM; hypergraph convolution; multi-branch short chain fusion; multi-level features; mural inpainting;
D O I
10.12454/j.jsuese.202200874
中图分类号
学科分类号
摘要
Aiming at the problems of low feature utilization and insufficient attention to context information in existing deep learning image restoration methods, a generative adversarial mural restoration model based on multi-level feature fusion and hypergraph convolution was proposed. Firstly, a pyramid feature extraction layer structure was designed, and the multi-scale feature extraction was carried out by using the pyramid feature layer. The mixed dilated convolution unit was used to expand the receptive field of multi-layer feature extraction, which overcomes the problem of insufficient feature extraction ability of single-scale convolution operation. Then, a multi-branch short chain fusion layer and a gating method were proposed to fuse the multi branch features, and the feature information between the adjacent branches was fused, so that the fused mural feature map has both the same branch features and the adjacent branch features, improving the utilization rate of feature information, and introducing the gating mechanism to select and fuse features to reduce the loss of detail information. Next, the fused features were passed through the ConvLSTM feature attention to enhance the attention to the mural context information. Finally, a hypergraph convolution mural long-range feature enhancement module was designed. By establishing a hypergraph convolution layer between the skip connection of the encoder and the decoder, the spatial feature information of the encoder was captured by hypergraph convolution and transferred to the decoder, which helpes the decoder to generate mural images and strengthenes the long-range dependence of features. Afterwards, the mural restoration was completed in a game against the SN–PatchGAN discriminator. Through the restoration experiments of digital Dunhuang murals, the results showed that the proposed method is superior to the comparative algorithms, and the restoration results of damaged murals are clearer and more natural. © 2024 Sichuan University. All rights reserved.
引用
收藏
页码:208 / 218
页数:10
相关论文
共 27 条
  • [1] Huan Wang, Qingquan Li, Qin Zou, Inpainting of Dunhuang murals by sparsely modeling the texture similarity and structure continuity[J], Journal on Computing and Cultural Heritage, 12, 3, (2019)
  • [2] Huan Wang, Qingquan Li, Sen Jia, A global and local feature weighted method for ancient murals inpainting[J], International Journal of Machine Learning and Cybernetics, 11, 6, pp. 1197-1216, (2020)
  • [3] Yuantao Chen, Haopeng Zhang, Linwu Liu, Et al., Research on image inpainting algorithm of improved total variation minimization method[J], Journal of Ambient Intelligence and Humanized Computing, 14, 5, pp. 5555-5564, (2023)
  • [4] Nianyi Wang, Weilan Wang, Wenjin Hu, Et al., Thanka mural inpainting based on multi-scale adaptive partial convolution and stroke-like mask[J], IEEE Transactions on Image Processing, 30, pp. 3720-3733, (2021)
  • [5] Ning Wang, Yipeng Zhang, Lefei Zhang, Dynamic selection network for image inpainting[J], IEEE Transactions on Image Processing, 30, pp. 1784-1798, (2021)
  • [6] Yong Chen, Yapeng Ai, Hongguang Guo, Inpainting algorithm for Dunhuang mural based on improved curvature-driven diffusion model[J], Journal of Computer-Aided Design & Computer Graphics, 32, 5, pp. 787-796, (2020)
  • [7] Abdulla A A, Ahmed M W., An improved image quality algorithm for exemplar-based image inpainting[J], Multimedia Tools and Applications, 80, 9, pp. 13143-13156, (2021)
  • [8] Zhiyuan Zha, Xin Yuan, Bihan Wen, Et al., Image restoration using joint patch-group-based sparse representation[J], IEEE Transactions on Image Processing, 29, pp. 7735-7750, (2020)
  • [9] Kawar B, Elad M, Ermon S, Et al., Denoising diffusion restoration models
  • [10] Ting Xu, Tingzhu Huang, Liangjian Deng, Et al., Exemplarbased image inpainting using adaptive two-stage structure-tensor based priority function and nonlocal filtering[J], Journal of Visual Communication and Image Representation, 83, (2022)