MRS-Net: an image inpainting algorithm with multi-scale residual attention fusion

被引:1
作者
Deng, Hongxia [1 ]
Qian, Guanyu [1 ]
Luo, Dongsheng [1 ]
Lv, Xindong [1 ]
Liu, Haoqi [2 ]
Li, Haifang [1 ]
机构
[1] Taiyuan Univ Technol, Coll Informat & Comp, Taiyuan, Peoples R China
[2] Univ Edinburgh, Sch Engn, Edinburgh, Midlothian, Scotland
基金
中国国家自然科学基金;
关键词
Residual sequence and extraction; Image inpainting algorithm; Generative adversarial network; Multi scale discriminator network;
D O I
10.1007/s10489-022-03866-w
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Image repair is to repair partially damaged images. At present, when repairing the images with arbitrary missing shape and the images with large defect area, the current methods have some problems, such as fuzzy image repair and differences in the repaired joints. Therefore, this paper proposes an image repair model of MRS-Net(Multiscale Residual Squeeze-and-congestion Networks) by using the generated countermeasure network to repair the images with arbitrary missing shape and large defect area. The generator adds a residual attention module in the connection layer between the encoder and the decoder to improve the repair ability of the model, and generates a network through multi-scale joint feedback against loss, reconstruction loss, perception loss, style loss and total variation loss, so as to ensure the visual consistency between the repair boundary and the surrounding real image. At the same time, the binary cross entropy loss feedback discriminant network is used. The proposed model is trained and tested on the dataset CelebA and Oxford buildings. Experiments show that the proposed model can effectively extract the missing information, Meanwhile, the repair results have natural transition boundaries and clear details. MRS-Net can improve SSIM(structural similarity) index by about 2% - 5% and PSNR(peak signal-to-noise ratio)index by about 1-3. The FID(Frechet Inception Distance score) straight index is reduced by 2-8, and the evaluation indexes are improved.when the missing area accounts for 5% - 10%, 11% - 20%, 21% - 30%, 31% - 40% and 41% - 50%,MRS-Net has better image restoration effect, which can reflect that MRS-Net has better robustness.The proposed image restoration method can repair faces, buildings and other scenes. At the same time, it can repair defects in different shapes and areas.
引用
收藏
页码:7497 / 7511
页数:15
相关论文
共 51 条
  • [1] Arjovsky M, 2017, PR MACH LEARN RES, V70
  • [2] PatchMatch: A Randomized Correspondence Algorithm for Structural Image Editing
    Barnes, Connelly
    Shechtman, Eli
    Finkelstein, Adam
    Goldman, Dan B.
    [J]. ACM TRANSACTIONS ON GRAPHICS, 2009, 28 (03):
  • [3] Image inpainting
    Bertalmio, M
    Sapiro, G
    Caselles, V
    Ballester, C
    [J]. SIGGRAPH 2000 CONFERENCE PROCEEDINGS, 2000, : 417 - 424
  • [4] Cao J., 2020, J COMPUT AIDED GRAPH, V32, P1315
  • [5] Nontexture inpainting by curvature-driven diffusions
    Chan, TF
    [J]. JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2001, 12 (04) : 436 - 449
  • [6] [常晨 Chang Chen], 2012, [计算机应用与软件, Computer Applications and Software], V29, P240
  • [7] [陈俊周 Chen Junzhou], 2019, [电子科技大学学报, Journal of University of Electronic Science and Technology of China], V48, P910
  • [8] Research on image Inpainting algorithm of improved GAN based on two-discriminations networks
    Chen, Yuantao
    Zhang, Haopeng
    Liu, Linwu
    Chen, Xi
    Zhang, Qian
    Yang, Kai
    Xia, Runlong
    Xie, Jingbo
    [J]. APPLIED INTELLIGENCE, 2021, 51 (06) : 3460 - 3474
  • [9] Demir U, 2018, ARXIV
  • [10] Ding Y., 2019, J DALIAN U NATL, V21, P458