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 条
  • [11] Yong Chen, Meifeng Tao, Jin Chen, Mural inpainting based on RPCA decomposition of block nuclear norm and entropy weighted clustering sparse representation[J], Journal of Harbin Institute of Technology, 53, 8, pp. 72-80, (2021)
  • [12] Jiahui Yu, Lin Zhe, Yang Jimei, Et al., Generative image inpainting with contextual attention[C], Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 5505-5514, (2018)
  • [13] Zefeng Yuan, Hengyu Li, Jingyi Liu, Et al., Multiview scene image inpainting based on conditional generative adversarial networks[J], IEEE Transactions on Intelligent Vehicles, 5, 2, pp. 314-323, (2020)
  • [14] Yuan Zeng, Yi Gong, Jin Zhang, Feature learning and patch matching for diverse image inpainting[J], Pattern Recognition, 119, (2021)
  • [15] Qiang Wang, Huijie Fan, Gan Sun, Et al., Recurrent generative adversarial network for face completion[J], IEEE Transactions on Multimedia, 23, pp. 429-442, (2021)
  • [16] Yanhong Zeng, Jianlong Fu, Chao Hongyang, Et al., Learning pyramid-context encoder network for high-quality image inpainting[C], Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR), pp. 1486-1494, (2019)
  • [17] Weize Quan, Ruisong Zhang, Yong Zhang, Et al., Image inpainting with local and global refinement[J], IEEE Transactions on Image Processing, 31, pp. 2405-2420, (2022)
  • [18] Nazeri K, Ng E, Joseph T, Et al., EdgeConnect:Generative image inpainting with adversarial edge learning[EB/OL]
  • [19] Yu Jiahui, Lin Zhe, Yang Jimei, Et al., Free-form image inpainting with gated convolution[C], Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision (ICCV), pp. 4470-4479, (2019)
  • [20] Chuanxia Zheng, Cham T J, Cai Jianfei, Pluralistic image completion[C], Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR), pp. 1438-1447, (2019)