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 条
[31]  
Ref Shu, 2013, Journal of Computer Aided Design & Computer Graphics, V25, P1682
[32]  
Rudin L. I., 1994, Proceedings ICIP-94 (Cat. No.94CH35708), P31, DOI 10.1109/ICIP.1994.413269
[33]  
[邵肖伟 Shao Xiaowei], 2004, [电路与系统学报, Journal of circuits and systems], V9, P113
[34]   PEPSI: Fast and Lightweight Network for Image Inpainting [J].
Shin, Yong-Goo ;
Sagong, Min-Cheol ;
Yeo, Yoon-Jae ;
Kim, Seung-Wook ;
Ko, Sung-Jea .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2021, 32 (01) :252-265
[35]  
van den Oord A, 2016, PR MACH LEARN RES, V48
[36]  
Wang, 2007, APPL IMAGE INPAINTIN
[37]  
Wang J., 2020, COMPUT TECHNOL DEV, V30, P24
[38]  
Wang N, 2019, PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P3748
[39]  
Weiyun P, 2020, RES IMAGE STYLE TRAN
[40]  
Xiao F., 2021, J XIAN UNIVER TECHNO, V41, P198