IEEE Access (Jan 2024)

Feature Consistency-Based Style Transfer for Landscape Images Using Dual-Channel Attention

  • Qiang Zhang,
  • Shuai Wang,
  • Dong Cui

DOI
https://doi.org/10.1109/ACCESS.2024.3485063
Journal volume & issue
Vol. 12
pp. 164018 – 164027

Abstract

Read online

With the rapid development of artificial intelligence technology, style transfer has become an important topic in current research. However, existing models are deficient in fusing content and style features, resulting in a large gap between the generated image and the target image. To solve this problem, we propose a feature-consistent landscape image style transfer model based on two-channel attention. Unlike traditional models that rely on VGG as a content encoder, VGG has a more limited detail extraction capability when dealing with high-resolution landscape images, so we introduce a Visual Transformer (VIT) to enhance the extraction of content features. In addition, by incorporating a channel attention mechanism in the latent space, we achieve consistency between content and style features, which in turn completes the alignment and fusion of feature distributions. Finally, contrast constraints are applied to accelerate the style transfer process. Comparison experiments show that our method outperforms other existing methods on the style transfer task.

Keywords