Inpainting algorithm based on hybrid dilated convolution and improved knowledge consistent attention

被引:0
作者
Li H. [1 ]
Yin H. [1 ]
Zhong X. [1 ]
Zhang Y. [1 ]
机构
[1] School of Information, Yunnan University, Kunming
来源
Huazhong Keji Daxue Xuebao (Ziran Kexue Ban)/Journal of Huazhong University of Science and Technology (Natural Science Edition) | 2023年 / 51卷 / 09期
关键词
attention mechanism; group normalization; hybrid dilated convolution; image inpainting; irregular hole;
D O I
10.13245/j.hust.238196
中图分类号
学科分类号
摘要
In order to effectively solve the problems of structural loss and local texture distortion in repairing large-scale and continuous holes in an image,an image restoration algorithm based on the improved knowledge consistent attention mechanism is proposed. Firstly,the partial convolution is performed to normalize the corrupted image and update the mask. Subsequently,the result is fed into the region identification module composed of two partial convolutions. Thereafter,the recognized information maps are sent into the feature induce structure containing the hybrid dilated convolution (HDC) and the improved knowledge consistent attention (KCA) mechanism. The features of the output images are combined after the induce iteration is completed.Finally,the reconstructed feature map is subjected to the post-processing by using deconvolution and de-residual bottleneck network to enhance the structural integrity of the repaired image.Group normalization (GN) are applied in the proposed module improve the convergence speed of the loss function. Experiments are fulfilled on public data set of Paris street view. The results demonstrate that the proposed algorithm can effectively inpaint large-scale and continuous damage in an image and avoid restoration distortion.Moreover,the proposed method demonstrates superior performance on the peak signal-to-noise ratio,structural similarity and time consuming compared with the state-of-the-art methods. © 2023 Huazhong University of Science and Technology. All rights reserved.
引用
收藏
页码:110 / 117
页数:7
相关论文
共 19 条
[1]  
BERTALMIO M,, SAPIRO G, CASELLES V, et al.Image inpainting[C], Proc of 27th Annual Conference on Computer Graphics and Interactive Techniques, pp. 417-424, (2000)
[2]  
586, 27, pp. 88-90, (2007)
[3]  
BARNES C, SHECHTMAN E, FINKELSTEIN A, PatchMatch: a randomized correspondence algorithm for structural image editing[J], ACM Trans Graph, 28, 3, pp. 1-11, (2009)
[4]  
YAN Z,, LI X, LI M, Shift-net:image inpainting via deep feature rearrangement[C], Proc of European Conference on Computer Vision (ECCV), pp. 1-17, (2018)
[5]  
YU J,, LIN Z,, YANG J, Generative image inpainting with contextual attention[C], Proc of 2018 IEEE/ CVF Conference on Computer Vision and Pattern Recognition, pp. 5505-5514, (2018)
[6]  
YANG C, LU X, LIN Z, High-resolution image inpainting using multi-scale neural patch synthesis [C], Proc of IEEE Conference on Computer Vision and Pattern Recognition, pp. 6721-6729, (2017)
[7]  
NAZERI K, NG E, JOSEPH T, Edgeconnect:structure guided image inpainting using edge prediction [C], Proc of IEEE International Conference on Computer Vision Workshops, pp. 3265-3274, (2019)
[8]  
YU J,, LIN Z,, YANG J, Free-form image inpainting with gated convolution[C], Proc of IEEE International Conference on Computer Vision, pp. 4471-4480, (2019)
[9]  
GUO Z, CHEN Z, YU T, Progressive image inpainting with full-resolution residual network[C], Proc of 27th ACM International Conference on Multimedia, pp. 2496-2504, (2019)
[10]  
LI J, WANG N, ZHANG L, Recurrent feature reasoning for image inpainting[C], Proc of IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7760-7768, (2020)