错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Image inpainting based on tensor ring decomposition with generative adversarial network

  • Jianjun Yuan,
  • Hong Wu,
  • Luoming Zhao,
  • Fujun Wu

摘要

Image inpainting is a fundamental task in the field of computer vision. However, there are three major challenges associated with this technique: (1) maintaining neighborhood texture consistency; (2) ensuring the rationality of visual structure; and (3) the limitation of existing inpainting models based on deep neural networks due to a large number of parameters. In order to tackle these challenges, we propose a novel generative adversarial network based on tensor ring decomposition to inpaint images with varying degrees of damage. First, we design a dual-path block that captures features at different scales without significant memory consumption. Every pair of dual-path blocks is incorporated into an enhanced residual module to integrate local and global features. Additionally, we propose the tensor ring layer to compress the convolution, reducing the number of model parameters and computational complexity. We then use a more accurate U-Net based discriminator to optimize the network by minimizing reconstruction loss, adversarial loss, perceptual loss and style loss. Extensive experiments demonstrate that, when compared with other state-of-the-art algorithms, our model shows superior performance in compression. The repaired images also exhibit reasonable texture structure and contextual semantic information.