Learning a Sketch Tensor Space for Image Inpainting of Man-made Scenes

被引:38
作者
Cao, Chenjie [1 ]
Fu, Yanwei [1 ]
机构
[1] Fudan Univ, Sch Data Sci, Shanghai, Peoples R China
来源
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021) | 2021年
关键词
NEURAL-NETWORKS;
D O I
10.1109/ICCV48922.2021.01424
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper studies the task of inpainting man-made scenes. It is very challenging due to the difficulty in preserving the visual patterns of images, such as edges, lines, and junctions. Especially, most previous works are failed to restore the object/building structures for images of man-made scenes. To this end, this paper proposes learning a Sketch Tensor (ST) space for inpainting man-made scenes. Such a space is learned to restore the edges, lines, and junctions in images, and thus makes reliable predictions of the holistic image structures. To facilitate the structure refinement, we propose a Multi-scale Sketch Tensor inpainting (MST) network, with a novel encoder-decoder structure. The encoder extracts lines and edges from the input images to project them into an ST space. From this space, the decoder is learned to restore the input images. Extensive experiments validate the efficacy of our model. Furthermore, our model can also achieve competitive performance in inpainting general nature images over the competitors.
引用
收藏
页码:14489 / 14498
页数:10
相关论文
共 49 条
[1]  
[Anonymous], 2016, CoRR. abs/1511.07122
[2]  
[Anonymous], 2015, A neural algorithm of artistic style, DOI DOI 10.1167/16.12.326
[3]  
[Anonymous], 2020, AAAI
[4]   PatchMatch: A Randomized Correspondence Algorithm for Structural Image Editing [J].
Barnes, Connelly ;
Shechtman, Eli ;
Finkelstein, Adam ;
Goldman, Dan B. .
ACM TRANSACTIONS ON GRAPHICS, 2009, 28 (03)
[5]  
CRIMINISI A, 2003, PROC CVPR IEEE, P721, DOI [DOI 10.1109/CVPR.2003.1211538, 10.1109/CVPR.2003.1211538]
[6]   Class-Balanced Loss Based on Effective Number of Samples [J].
Cui, Yin ;
Jia, Menglin ;
Lin, Tsung-Yi ;
Song, Yang ;
Belongie, Serge .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :9260-9269
[7]  
Denis P, 2008, LECT NOTES COMPUT SC, V5303, P197, DOI 10.1007/978-3-540-88688-4_15
[8]   On the Canny edge detector [J].
Ding, LJ ;
Goshtasby, A .
PATTERN RECOGNITION, 2001, 34 (03) :721-725
[9]   USE OF HOUGH TRANSFORMATION TO DETECT LINES AND CURVES IN PICTURES [J].
DUDA, RO ;
HART, PE .
COMMUNICATIONS OF THE ACM, 1972, 15 (01) :11-&
[10]   Image Inpainting: A Review [J].
Elharrouss, Omar ;
Almaadeed, Noor ;
Al-Maadeed, Somaya ;
Akbari, Younes .
NEURAL PROCESSING LETTERS, 2020, 51 (02) :2007-2028